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=

你可能感兴趣的:(YOLOv8,YOLO5系列改进专栏,python,机器学习,深度学习,YOLOv8)