以检测行人为例,找到object_detection/utils/visualization_utils.py
打开后,在大约735行可以看到如下代码段
修改为
即在顶部增加两行
display_str_list = box_to_display_str_map[box]
if (("person") in display_str_list[0]):
此处需要检测哪一类物体就改为该label即可
、mark一下 感谢https://blog.csdn.net/sinat_39389724/article/details/89643774