我正在使用 Tensorflow 对象检测 API 来训练我自己的车辆检测器。当我使用 Object detection tutorial 测试我的模型时,我发现在某些情况下,卡车被检测为汽车和卡车,周围有两个重叠的边界框。我只想留下检测分数最高的那个。
我知道对象检测 API 确实会消除重叠的边界框,但它不会针对跨不同类的边界框这样做。有没有办法摆脱重叠的盒子?我可以更改对象检测 API 代码中的任何地方来实现该目标吗?
最佳答案
有两种方法可以消除重叠的边界框。
1、在文件visualization_utils.py中的“visualize_boxes_and_labels_on_image_array”函数中将“min_score_thresh”参数设置得更大;
2、在配置文件中设置非最大抑制阈值
关于python - 摆脱 Tensorflow Object Detection API 中不同类别的重叠边界框,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49951422/