python - tf.boolean_mask 不接受轴参数

Question

这是我的代码：

 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)

错误，我得到：

TypeError: boolean_mask() 得到了一个意外的关键字参数“轴”

不接受轴参数，tf.boolean_mask()但可以在文档中看到一个有效的参数：https ://www.tensorflow.org/api_docs/python/tf/boolean_mask

标签： pythontensorflowkerasyolo

Answer 1

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