TensorFlow 对象检测限制

Question

按照这里的代码：https://github.com/tensorflow/models/blob/master/research/object_detection/object_detection_tutorial.ipynb

无论输入什么图像，检测到的物体似乎都存在 20 个硬性限制。示例：

这个问题也出现在这个帖子里：TensorFlow Object Detection API Weird Behavior

是否有一些配置或参数可以更改以增加检测到的对象数量？

编辑：我可以确认检测到超过 20 个对象，但最终输出中最多显示 20 个。有没有办法提高这个限制？

Answer 1

可以在配置文件中设置最大检测数。默认情况下它通常是 300，所以你应该没问题。

您的问题是 显示 检测的数量。在您的代码结束时，您可以致电vis_util.visualize_boxes_and_labels_on_image_array。只需在其参数中添加max_boxes_to_draw=None 即可显示所有检测（或者如果需要，可以选择更大的数字）。