YOLOv5及V8改进系列:替换IOU损失函数,SIoU、Wise-IoU等多种IoU一键替换
YOLOv5及V8改进系列-替换训练IoU损失函数
- 一、在YoloV5中添加EIoU,SIoU,AlphaIoU,FocalEIoU,Wise-IoU.
-
- 1. YOLOv5替换Wise-IoU:
- 二、改进YOLOv8
- 建议
一、在YoloV5中添加EIoU,SIoU,AlphaIoU,FocalEIoU,Wise-IoU.
1. YOLOv5替换Wise-IoU:
YOLOv5中默认采用的是CIoU,
替换文件: 在YOLOv5/utils/loss.py,函数名为:bbox_iou。
替换方式: 有GIoU,DIoU,CIoU的bool参数可以选择,如果全部为False的时候,其会返回最普通的Iou,如果其中一个为True的时候,即返回设定为True的那个Iou。
修改1
def bbox_iou(box1, box2, xywh=True, GIoU=False, DIoU=False, CIoU=False, SIoU=