YOLOv5的 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个甜甜圈，将一个甜甜圈检测成多个甜甜圈。
-
参考：
