YOLOv4 同一物体的多个边界框答案

【问题标题】：YOLOv4 multiple bounding box for the same objectYOLOv4 同一物体的多个边界框
【发布时间】：2021-09-11 14:05:22
【问题描述】：

我已经使用暗网实现了 YOLOv4 对象检测。训练结束后，我开始使用以下命令在视频上测试模型： !./darknet detector demo cfg/coco.data cfg/yolov4.cfg yolov4.weights -dont_show test.mp4 -i 0 -out_filename results.avi 有时，在同一个对象上绘制了多个边界框。例如，在这张图片example_image 中，我有两个边界框。我怎样才能只显示一个概率最高的边界框。我尝试实现 IOU 但失败了，尤其是暗网包含多个文件，我不知道要修改哪个文件。我尝试修改 darknet_video.py 但没有成功。 PS：我尝试更改阈值，但此问题仍然存在。有没有人可以解决这个问题。

【问题讨论】：

标签： object-detection yolo bounding-box darknet

【解决方案1】：

是的，这很正常。更准确地说，YOLOv4 应该为每个图像提供多个边界框，因为有多个锚点和多个尺度。对于这个 YOLO 使用 Greedy-NMS（或 hard-nms）。它会丢弃置信度较低的框和一个对象上的多个框，因为它们的 IOU 会很高。您应该增加 conf_treshold 和 iou_treshold，例如0.5 通过在命令中包含 thresh 标志。

!./darknet detector demo cfg/coco.data cfg/yolov4.cfg yolov4.weights -dont_show test.mp4 -i 0 -out_filename results.avi -thresh 0.5

【讨论】：