YOLOv8 修改 IoU(CIOU, DIOU, SIOU, EIOU, Focal_EIOU) YOLOv8 obb修改CIOU

1、YOLO v8版本

需要注意的是,现阶段yolo v8的代码几乎每天都在变动,现在日期是24年4月14日,因此我的代码改动也是基于这几天最新的yolo v8版本的。

2、IOU

目前版本的Yolo v8已经内置了多个不同的IOU,可以直接进行调用,至于具体使用哪一个,需要你根据自己的情况进行选择。

3、修改IOU

修改IOU非常简单,我们只需要改两个地方的参数即可。首先是ultralytics/utils/loss.py,在该文件的75行左右有下面一行代码:

iou = bbox_iou(pred_bboxes[fg_mask], target_bboxes[fg_mask], xywh=False, EIoU=True)

你需要改的地方就是把你想要的IOU进行替换即可,例如 DIOU = Ture

第二个要修改的地方在ultralytics/utils/tal.py文件125行左右

return bbox_iou(gt_bboxes, pd_bboxes, xywh=False, EIoU=True).squeeze(-1).clamp_(0)

修改方式如上,进行替换即可,不过需要说明的是,这里所替换的IOU都是在ultralytics/utils/metrics.py文件里定义的,你可以在1326行左右找到定义的函数,如下:

只要这里面有的IOU,你都可以在上面的两个地方进行修改使用。

4、旋转框IOU

修改上述两个地方的IOU是不能用于旋转框的,用于旋转框的IOU,目前这个版本的yolo v8只有CIOU可用,同样,在ultralytics/utils/loss.py文件中我们找到RotatedBboxLoss(BboxLoss)这个类,大概在118行处有一行这样的代码

iou = probiou(pred_bboxes[fg_mask], target_bboxes[fg_mask],CIoU=True)

我们将其CIOU=Ture。

在ultralytics/utils/tal.py文件找到class RotatedTaskAlignedAssigner(TaskAlignedAssigner)函数大概在265行左右修改代码为:

return probiou(gt_bboxes, pd_bboxes,CIoU=True).squeeze(-1).clamp_(0)

如果大家有旋转框的其他IOU代码可以使用,欢迎留言

  • 10
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值