在yolov8中更改损失函数

第一步，打开需要修改的文件

首先，在ultralytics/utils文件中打开，loss.py和tal.py这两个文件

然后，在loss.py文件中找到 class BboxLoss(nn.Module)函数，再找到bbox_iou,按住Ctrl键后进入

此时会进入

函数中的Giou，Diou，Ciou就是损失函数，eps为精度

第二步，修改文件

首先将def bbox_iou函数注释掉，然后在后面添加如下代码

class WIoU_Scale:
    ''' monotonous: {
            None: origin v1
            True: monotonic FM v2
            False: non-monotonic FM v3
        }
        momentum: The momentum of running mean'''

    iou_mean = 1.
    monotonous = False
    _momentum = 1 - 0.5 ** (1 / 7000)
    _is_train = True

    def __init__(self, iou):
        self.iou = iou
        self._update(self)

    @classmethod
    def _update(cls, self):
        if cls._is_train: cls.iou_mean = (1 - cls._momentum) * cls.iou_mean + \
                                         cls._momentum * self.iou.detach().mean().item()

    @classmethod
    def _scaled_loss(cls, self, gamma=1.9, delta=3):
        if isinstance(self.monotonous, bool):
            if self.monotonous:
                return (self.iou.detach() / self.iou_mean).sqrt()
            else:
                beta = self.iou.detach() / self.iou_mean
                alpha = delta * torch.pow(gamma, beta - delta)
                return beta / alpha
        return 1


def