关于读取模型权重的问题

taohan10200 / IIM

PyTorch implementations of the paper: "Learning Independent Instance Maps for Crowd Localization"

MIT License

163 stars 39 forks source link

Open Veefas opened 3 years ago

Veefas commented 3 years ago

@taohan10200 ，作者你好，我按照readme的训练过程指导在NWPU数据集上迭代训练时，模型按要求这样保存与加载，但预测出来的像素点始终是0。使用你提供的权重，上述模型加载方式就没有问题，调试也没能找到方向，可否麻烦你帮忙看看，感谢。

gjy3035 commented 3 years ago

strict=True 才能保证参数被完全加载。如果改为True报错，可以看看这个问题 https://github.com/taohan10200/IIM/issues/5

Veefas commented 3 years ago

strict=True 才能保证参数被完全加载。如果改为True报错，可以看看这个问题

5

我是用4核GPU训练的，test的时候试了4核和1核，strict改为True以后还是不行，预测的像素点全是0。。按理说权重加载正确，效果应该和验证集差不多，还是有些费解

taohan10200 commented 3 years ago

您好，我们刚才已经在SHHB数据集上重新验证了，对于训练过程中保存的模型是可以直接使用test.py进行预测的，请问您是不是clone的最新版本，以及是不是在后续改动过程中引起了新的问题？训练过程中保存的模型：

在test.py中添加要测试的模型路径，以及修改数据集名称等相关参数：在测试集上的预测结果：./saved_exp_results/SHHB_HR_Net_test.txt'

Veefas commented 3 years ago

您好，我早上用SHHB数据集上训练验证，的确没有问题。但之前那一批训练的结果就是预测像素点为0，十分的迷幻，应该是我改动了哪个地方导致的，暂时不纠结这个点了，anyway，感谢作者大大！