object - 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 添加到它的参数中即可显示所有检测(或者如果需要，可以选择更大的数字)。