search

Krasjet-Yu / YOLO-FaceV2

YOLO-FaceV2: A Scale and Occlusion Aware Face Detector
163 stars 24 forks source link

repulsion loss问题 #10

Open SSF101 opened 1 year ago

SSF101 commented 1 year ago

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

SSF101 commented 1 year ago

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

Krasjet-Yu commented 1 year ago

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

SSF101 commented 1 year ago

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

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

liiicon commented 7 months ago

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

wujians122 commented 5 months ago

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