YOLOV7改进-Soft-NMS(支持多种IoU变种选择)

soft-nmslink
1、复制到utils-general.py下
2、找到nms的代码,在它前面复制过来就行
YOLOV7改进-Soft-NMS(支持多种IoU变种选择)_第1张图片
3、在之前的nms模块下修改,注释
YOLOV7改进-Soft-NMS(支持多种IoU变种选择)_第2张图片

4、选择iou的地方,默认正常的
5、xx=true即可
YOLOV7改进-Soft-NMS(支持多种IoU变种选择)_第3张图片
6、训练的时候开启,会增加训练成本
7、在test的时候用
8、密集预测时精度降低

你可能感兴趣的:(yolov7,YOLO)