YOLOv5的 detect.py 参数解读

目录

• YOLOv5/v7 detect.py 传参

 

---

回到顶部

YOLOv5/v7 detect.py 传参

• conf_thres

  • Confidence Threshold，置信度阈值。
  • 只显示预测概率超过conf_thres的预测结果。
    • 想让YOLO只标记可能性高的地方，就把这个参数提高。

• iou_thres

  • Intersect over Union Threshold，交并比阈值。
  • IOU值：预测框大小 ∩ 真实框大小 / 预测框大小 ∪ 真实框大小。
    • 预测框与真实框的交集与并集的取值。
  • iou_thres在detect.py中：

    • 越大，则容易将对于同一个物品的不同预测结果 当成 对多个物品的多个预测结果，导致一个物品出现了多个预测结果。

    • 越小，则容易将对于多个物品的不同预测结果 当成 对同一个物品的不同预测结果，导致多个物品只出现了一个预测结果。

    • 如下面对甜甜圈的预测一般，多个甜甜圈重叠在一起的图片中：

      • 将iou阈值设为0.65，检测出了18个甜甜圈。
      • 设为0，检测出了6个甜甜圈，并且检测区域都没有重复。
      • 设为0.98，检测出了111个甜甜圈，将一个甜甜圈检测成多个甜甜圈。

参考：

https://mwhls.top/2428.html