pytorch-YOLO-v1 icon indicating copy to clipboard operation
pytorch-YOLO-v1 copied to clipboard

Published 20 hours ago • verified publisher icon abeardear

损失函数的参数好像有点问题

Open zust-pss opened this issue 3 years ago • 4 comments

(self.l_coord * loc_loss + 2 * contain_loss + not_contain_loss + self.l_noobj * nooobj_loss + class_loss) / N

contain_loss前系数为何是2？ not_contain_loss前应该有self.l_noobj作为系数

zust-pss avatar Oct 26 '21 07:10 zust-pss

请问明白了么？我刚好也在纠结 为何前面系数 2

punk1 avatar Mar 18 '22 12:03 punk1

按照论文应该没有2的，还有就是为啥有not_contain_loss这个东西，不就是nooobj_loss吗？

myshrry avatar Jun 23 '22 19:06 myshrry

作者的loss跟论文不一样，我建议如果要复现的话，可以自己重新写一下

Vandaci avatar Jul 20 '22 23:07 Vandaci

我认为应该是作者在实际调参的时候通过测试和训练发现模型预测出来的置信度过低或contain loss过高 所以调整了权重参数以获得更好的检测效果

xiazhi1 avatar Mar 05 '24 12:03 xiazhi1