在train.py里 按Ctrl+F键,
查询
with torch.cuda.amp.autocast(amp):
下面有个compute_loss
按住Ctrl点compute_loss
就会进入新页码
再按住Ctrl点Computeloss(这里是等号的右边那个)
进入新界面
往下找有个
或者按Ctrl+F,搜
iou = bbox_iou
这里显示的yolov5默认的IOU为CIOU
去更换IOU,yolov5自带有几个
按住Ctrl单点bbox_iou
会进入新的页面
这里就是几个自带的IOU
若要替换的话,比如使用Giou
就在这里,把CIoU换成GIoU就可以了