其他分享
首页 > 其他分享> > (转)YOLOv5的置信度阀值与iou阀值

(转)YOLOv5的置信度阀值与iou阀值

作者:互联网


conf_thres

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

iou_thres

  1. Intersect over Union Threshold,交并比阈值。
  2. IOU值:预测框大小∩真实框大小 / 预测框大小∪真实框大小。(预测框与真实框的交集与并集的取值。)
  3. iou_thres在detect.py中:

图1:iou-thres = 0.65   (将iou阈值设为0.65,检测出了18个甜甜圈。)

 

 

 

图2:iou-thres = 0  (设为0,检测出了6个甜甜圈,并且检测区域都没有重复。)

 

 

 

图3:iou-thres = 0.98

 

 

 

PR图

AP

mAP

如何提高mAP
   优化数据集:
      YOLO官方推荐数据集需求:
        每类图片数量:每类大于1500张图片
        每个类的实例:每类实例(标签)大于一万份。
        图像多样性:确保图片来自不同环境。对于现实世界物体的识别,推荐使用来自不同时间、不同季节、不同天气、不同光照度、不同角度、不同来源的图片。
        标签一致性:所有类中所有实例必须都被标签标记,不要遗漏。
        标签准确性:标签能准确标记对象,不要在对象与标记框之间留下缝隙。没有对象可以不用标记(背景图片)。
        背景图片:背景图片是指没有对象的图片,用来减少误报。YOLO推荐0~10%的背景图片来减少误报(COCO数据集包含1000张背景图片,占据总数1%)。
简单来说:
   增加不同角度/光照度的数据集。
   增加不同环境的数据集。
   增加背景图片(无目标的图片)。
   增加数据集大小。
   确保标签标定准确。


————————————————
版权声明:本文为CSDN博主「MWHLS」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/asd123pwj/article/details/117171868

标签:mAP,YOLOv5,置信度,预测,标签,iou,背景图片,thres,阀值
来源: https://www.cnblogs.com/answerThe/p/15616811.html