tf.boolean_mask不接受axis参数

时间:2019-03-03 07:32:33

标签: python tensorflow keras yolo

这是我的代码:

public class MainActivity extends AppCompatActivity {

    DatabaseReference dref;
    StorageReference sref;

    public static MyAppAdapter myAppAdapter;
    public static ViewHolder viewHolder;
    private ArrayList<Data> array;
    private SwipeFlingAdapterView flingContainer;

    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.activity_main);

        flingContainer = (SwipeFlingAdapterView) findViewById(R.id.frame);

        array = new ArrayList<>();

        myAppAdapter = new MyAppAdapter(array, MainActivity.this);
        flingContainer.setAdapter(myAppAdapter);

        dref = FirebaseDatabase.getInstance().getReference().child("root");

        dref.addValueEventListener(new ValueEventListener() {
            @Override
            public void onDataChange(@NonNull DataSnapshot dataSnapshot) {

                for(DataSnapshot snap: dataSnapshot.getChildren()){
                    firedb firedata = snap.getValue(firedb.class);
                    array.add(new Data(firedata.getImgurl(),firedata.getDesc()));
                    Toast.makeText(MainActivity.this, "Data.desc: "+firedata.getDesc(), Toast.LENGTH_SHORT).show();
                }
            }

            @Override
            public void onCancelled(@NonNull DatabaseError databaseError) {

            }
        });




        array.add(new Data("http://www.androidtutorialpoint.com/wp-content/uploads/2016/11/Katrina-Kaif.jpg", "Hi I am Katrina Kaif. Wanna chat with me ?. \n" +
                "Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat. Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur. Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum."));

错误,我得到:

  

TypeError:boolean_mask()获得了意外的关键字参数'axis'

44 scores = tf.boolean_mask(box_class_scores,filtering_mask,axis=-1) 45 boxes = tf.boolean_mask(boxes,filtering_mask,axis=-1) 46 classes = tf.boolean_mask(box_classes,filtering_mask,axis=-1) 不接受轴参数,但是它是有效的参数,如文档中所示:https://www.tensorflow.org/api_docs/python/tf/boolean_mask

enter image description here

2 个答案:

答案 0 :(得分:1)

如果从函数中删除axis参数,它将起作用。您遇到的错误将得到解决。

答案 1 :(得分:0)

正如@syltruong所指出的,请检查您所使用的Keras版本是否与文档相匹配。现在,文档使用版本2.2.4。

在旁注@Ganesh和@Sushant中,您实际上不需要在Deep Learning专业化课程中为特定代码使用axis参数。