dupc2018
dupc2018
训练测试的embt都是你原来的都没动,也就是数据集换了一下,其他的都没变,结果输出的7帧全部都是一样的,而且结果不怎么清晰,不知道原因出现在什么地方了
直接使用train_gopro.py加载GoPro数据集是不是有点问题呢,我使用自己的数据集用train_gopro.py训练的,测试是用demo_8x.py脚本测试的,造成输出的都一样,连输出的图片文件大小都一样
是不是demo_8x.py的原因呢,写上False 加载模型的时候 model.load_state_dict(torch.load('/mnt/IFRNet-main/checkpoint/IFRNet_L/2022-06-26 02:14:46/IFRNet_L_latest.pth'),False),但是如果不写False就报错 RuntimeError: Error(s) in loading state_dict for Model: Missing key(s) in state_dict: "encoder.pyramid1.0.0.weight", "encoder.pyramid1.0.0.bias", "encoder.pyramid1.0.1.weight", "encoder.pyramid1.1.0.weight", "encoder.pyramid1.1.0.bias", "encoder.pyramid1.1.1.weight", "encoder.pyramid2.0.0.weight", "encoder.pyramid2.0.0.bias", "encoder.pyramid2.0.1.weight", "encoder.pyramid2.1.0.weight", "encoder.pyramid2.1.0.bias", "encoder.pyramid2.1.1.weight", "encoder.pyramid3.0.0.weight", "encoder.pyramid3.0.0.bias", "encoder.pyramid3.0.1.weight",...
Using a paired dataset, training the recurrent network was found to be very close rec_B to the real_B, and the fake_B was much different from both rec_B and real_B