yolov5中v4.0版本版本如何替换为DIou_nms,以及更改损失函数为ciou

Question

yolov5中v4.0版本版本如何替换为DIou_nms,以及更改损失函数为ciou

Answer 1

#utils/general.py
def bbox_iou(box1, box2, x1y1x2y2=True, GIoU=False, DIoU=False, CIoU=False, eps=1e-9):

可以看看这个函数的接口。所以有两种方法，一种是在调用这个bbox_iou函数的位置里面加上参数，像这样：

bbox_iou(xxxxxx,CIoU=True)

还有一种就是直接强制让其他的iou全部为False，CIoU=True

def bbox_iou(box1, box2, x1y1x2y2=True, GIoU=False, DIoU=False, CIoU=False, eps=1e-9):
    GIoU=False
    DIoU=False
    CIoU=True