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代码可以使用，欢迎留言