YOLO-FaceV2 icon indicating copy to clipboard operation
YOLO-FaceV2 copied to clipboard

repulsion loss问题

Open SSF101 opened this issue 2 years ago • 5 comments

我尝试将您提供的代码中的repulsion loss 迁移到我所用的yolov5-6.0项目中,但是出现了rep loss不参与训练的问题,每训练完一个epoch得到的各项损失及map指标都与未更改之前相同。我确保在前向传播过程中rep loss参与进来了,因为可以打印出rep损失,并且每个epoch的训练时间增加了很多,请问这是什么问题。

SSF101 avatar Aug 13 '22 00:08 SSF101

将yolofacev2在torch1.12上进行实验(torch1.12通过固定随机种子可以保证相同参数下训练完每个epoch得到的结果完全相同),我发现注释掉Replusion loss部分和不注释Replusion 得到的结果是相同的,请问这是一个bug吗?

SSF101 avatar Aug 14 '22 15:08 SSF101

您说的结果是指widerface的val数据集的mAP吗?

Krasjet-Yu avatar Aug 15 '22 01:08 Krasjet-Yu

您说的结果是指widerface的val数据集的mAP吗?

不是,是在其他数据集上的结果,这个损失和数据集相关吗

SSF101 avatar Aug 15 '22 06:08 SSF101

请问你解决了嘛,我在yolov7上移植了repulsion loss,和你的情况一样,前后训练值几乎一样,但是rep loss确实是参与前向传播了的 image

liiicon avatar Nov 22 '23 12:11 liiicon

我觉得是reploss占loss的比例太小了,默认参数lrep = self.hyp['alpha'] * lrepGT / 3.0 + self.hyp['beta'] * lrepBox / 3.0,这里给的是3.0,默认的是0.5

wujians122 avatar Jan 04 '24 12:01 wujians122