search

zychen-ustc / PSD-Principled-Synthetic-to-Real-Dehazing-Guided-by-Physical-Priors

Zeyuan Chen, Yangchao Wang, Yang Yang and Dong Liu. "PSD: Principled Synthetic-to-Real Dehazing Guided by Physical Priors". IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2021.
MIT License
111 stars 18 forks source link

你提供的预训练模型是什么？ #3

Closed mengxzh15 closed 2 years ago

mengxzh15 commented 2 years ago

你提供的PSD-MSBDN,PSD-GCA,PSD-FFA是你微调之后的结果还是ＭＳＢＤＮ，ＧＣＡ等原来模型中的权重呢？

mengxzh15 commented 2 years ago

你的代码里是这个torch.save(net.state_dict(), '/output/{}_epoch{}.pth'.format(opt.backbone, epoch))，与提供模型名字　不统一，让人很疑惑

zychen-ustc commented 2 years ago

您好，抱歉回复的稍晚了。请参考readme，使用test.py进行测试。提供的model都是微调后的最终结果。而您提供的代码应该是finetune.py文件中的，这个文件是用于fine-tune的，这里load的模型应该是使用synthetic data训练出的pre-train model。