张量流对象检测限

Question

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

无论输入的图像如何,都检测到20个物体的硬限制.例:

在这篇文章中也可以看到这个问题:TensorFlow对象检测API怪异行为

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

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

Answer 1

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

这里的问题是显示的检测数.在您的代码结束时,您可以打电话给vis_util.visualize_boxes_and_labels_on_image_array.只需添加max_boxes_to_draw=None其参数即可显示所有检测项(或根据需要选择更大的数字).