DBCNN-PyTorch

DBCNN-PyTorch copied to clipboard

如何重新训练scnn

3

你好，最近我根据您论文中的思路和github代码进行了复现。目前想尝试修改SCNN架构，但是缺少对应的数据集，查阅了相关的代码https://github.com/zwx8981/DBCNN，由于能力有限，没看懂如何生成训练SCNN预训练模型的数据集（包括失真图片和label），您能否提供python版本生成预训练数据集的代码，不胜感激。 还有一个问题，对于DBCNN的优化，您是否有一些思路。

hadesfgh

指向/DBCNN/pretrained_scnn/scnn.pkl 时发生错误：  

Jiyang-Liu0

请问在训练过程中保存下来的模型该如何加载呢？我在试图使用训练好的模型进行评测时遇到了参数不对应的问题，报错如下： File "/DBCNN-PyTorch-master/DBCNN.py", line 53, in __init__ scnn.load_state_dict(torch.load(scnn_root)) File "/home/guest1/anaconda3/envs/wpf/lib/python3.10/site-packages/torch/nn/modules/module.py", line 2153, in load_state_dict raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format( RuntimeError: Error(s) in loading state_dict for DataParallel: Missing key(s)...

charpng