cheng-Lad
2021-07-16 11:26
采纳率: 100%
浏览 497

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

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

  • 写回答
  • 关注问题
  • 收藏
  • 邀请回答

2条回答 默认 最新

  • 爱晚乏客游 2021-07-16 13:20
    已采纳
    #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
    
    已采纳该答案
    3 打赏 评论
  • 啥也不懂的研究生 2021-10-14 21:58

    非极大值抑制怎么改为DIOU_nms,解决了吗

    打赏 评论

相关推荐 更多相似问题