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个甜甜圈,将一个甜甜圈检测成多个甜甜圈。
-
参考: