张量流对象检测限

Posted 2021-04-01

遵循以下代码：https://github.com/tensorflow/models/blob/master/research/object_detection/object_detection_tutorial.ipynb

无论输入的图像如何，都检测到20个物体的硬限制。例如：

问题也出现在这篇文章中：TensorFlow Object Detection API Weird Behavior

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

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

答案

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

这里的问题是显示的检测数。在代码结束时，您可以调用vis_util.visualize_boxes_and_labels_on_image_array。只需将max_boxes_to_draw=None添加到其参数中即可显示所有检测项（或根据需要选择更大的数字）。