yolov4-pytorch
yolov4-pytorch copied to clipboard
bubbliiiing
单类别小目标检测在yolo.py文件中修改confidence,nms_iou指标对map,recall,precision结果没有作用？
单类别小目标检测在yolo.py文件中修改confidence,nms_iou值对map,recall,precision结果没有作用？是confidence,nms_iou值不在yolo.py文件里面改吗？我看utitle.box.py和getmap.py文件都有 ，utitle.box.py改了confidence,nms_iou也没用，只有getmap.py文件改了confidence=0.0001,nms_iou=0.01精度提高了一个点但是precision和recall指标一点变化都没
看一下注释 感谢大佬，我在3.1版本看到了，你的 utils文件夹utitle.box.py 中def non_max_suppression(self, prediction, num_classes, input_shape, image_shape, letterbox_image, conf_thres=0.5, nms_thres=0.4): 。 yolo.py中"confidence" : 0.5, #---------------------------------------------------------------------# # 非极大抑制所用到的nms_iou大小 #---------------------------------------------------------------------# "nms_iou" : 0.3, getmap.py中 confidence = 0.001 #--------------------------------------------------------------------------------------# # 预测时使用到的非极大抑制值的大小，越大表示非极大抑制越不严格。 #
# 该值一般不调整。 #--------------------------------------------------------------------------------------# nms_iou = 0.5 那 confidence，nms_iou是getmap.py为准吗？我准备在这里面改