YOLO-FaceV2
YOLO-FaceV2 copied to clipboard
repulsion loss问题
我尝试将您提供的代码中的repulsion loss 迁移到我所用的yolov5-6.0项目中,但是出现了rep loss不参与训练的问题,每训练完一个epoch得到的各项损失及map指标都与未更改之前相同。我确保在前向传播过程中rep loss参与进来了,因为可以打印出rep损失,并且每个epoch的训练时间增加了很多,请问这是什么问题。
将yolofacev2在torch1.12上进行实验(torch1.12通过固定随机种子可以保证相同参数下训练完每个epoch得到的结果完全相同),我发现注释掉Replusion loss部分和不注释Replusion 得到的结果是相同的,请问这是一个bug吗?
您说的结果是指widerface的val数据集的mAP吗?
您说的结果是指widerface的val数据集的mAP吗?
不是,是在其他数据集上的结果,这个损失和数据集相关吗
请问你解决了嘛,我在yolov7上移植了repulsion loss,和你的情况一样,前后训练值几乎一样,但是rep loss确实是参与前向传播了的
我觉得是reploss占loss的比例太小了,默认参数lrep = self.hyp['alpha'] * lrepGT / 3.0 + self.hyp['beta'] * lrepBox / 3.0,这里给的是3.0,默认的是0.5