关于训练模型

[1529046970]

您好！请问您提供的pyramidbox.pth是通过您给的代码训练出来的吗？我用您的代码训练了模型，在测试的时候每张图片得到的结果都是如下图左所示（最后的图是用您提供的模型测试出的）。我是否是在训练的某一步骤出了错呢？谢谢！

[yxlijun]

是用我提供的代码训练出来的，你提供的的图我看着没有问题啊

[1529046970]

就是两张图都是同样图片得到的结果。第一张图我自己跑出来的是750（根据您的模型跑出来的是70），最后一列分数，我的值都是0.99+（您的是前面的是0.9+，后面变小了）。非常感谢！

[yxlijun]

你的750个数据都是0.99+么

[1529046970]

嗯嗯

[1529046970]

就是训练出来后有很多的.pth模型文件，我用了好几个试了，结果都是这样～是否我训练步骤出了问题呢？

[yxlijun]

应该是哪出了问题，是直接按照我的训练代码跑的吗，有没有改什么参数，loss有下降吗，你用的.pth是多少次迭代之后的？

[1529046970]

参数没有改过，预训练模型是您提供的VGG.pth用了120000和最后一次，也试了val()中保存的pyramidbox.pth。

[yxlijun]

你应该用了多GPU训练是吧

[1529046970]

没有用多GPU，那个电脑就一个显卡～

[yxlijun]

这不应该啊，我在我这边训练都没有问题，你是否测试过demo.py,效果怎么样

[1529046970]

我还没测试过demo.py 我再测试一下。应该我哪里出了问题，我再找一找。非常感谢您！

[Llq201809]

请问，这个代码训练时使用了验证集，测试集标注不公开，所以我还是用的验证集去测试，但是效果不好。您们是怎么测试的widerface呢？ @1529046970 @yxlijun