YOLOv10模型改进-损失函数改进-第85篇：YOLOv10改进策略【损失函数】| EIoU损失函数

一、本文介绍

本文记录的是利用EIoU（Enhanced IoU）损失函数改进YOLOv10的定位精度。EIoU在CIoU基础上进一步优化了形状损失。

二、EIoU损失函数介绍

2.1 设计出发点

CIoU的形状损失不够精确，EIoU将形状损失分解为宽度和高度两个方向，实现更精准的形状匹配。

2.2 损失函数公式

EIoU损失：

EIoU = IoU - (ρ²(b, b_gt) / c²) - (ρ_w² / c_w²) - (ρ_h² / c_h²)

其中：

• ρ_w²：宽度方向距离平方
• ρ_h²：高度方向距离平方
• c_w²：宽度方向包围框距离平方
• c_h²：高度方向包围框距离平方

三、EIoU的实现代码

import torch
import torch.nn as nn