在 YOLO Fastest(或类似 YOLO 架构)模型的推理阶段,置信度阈值(conf-thres)
和NMS IoU 阈值(iou-thres)
是两个关键后处理参数,其推荐设置如下:
参数 | 推荐值 | 说明 |
---|---|---|
置信度阈值 | 0.25 ~ 0.4 | 用于过滤低置信度框,0.25 是 YOLOv5 的默认值,0.4 是平衡精度和召回率的常用值。 |
NMS IoU 阈值 | 0.45 ~ 0.5 | 用于抑制重叠框,0.45 是 YOLOv5 的默认值,0.5 是多数检测任务的标准选择。 |
举例配置(YOLOv5/YOLOFast通用):
conf_thres = 0.25 # 或 0.4(更严格)
iou_thres = 0.45 # 或 0.5(更严格)
注意事项:
- 密集目标(如人群):可适当降低
iou_thres
至 0.3~0.4,避免漏检。 - 高召回率需求:降低
conf_thres
至 0.1~0.2,但会增加误检。 - 高精度需求:提高
conf_thres
至 0.5~0.6,但可能漏检模糊目标。
根据实际数据验证调整,通常从默认值开始微调。