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
115 stars 19 forks source link

关于finetune #17

Closed ZekaiShao25 closed 2 years ago

ZekaiShao25 commented 2 years ago

感谢您能为此出色的工作提供开源代码和微调后的模型！ 我对微调的无监督过程中各个损失函数的设置比较感兴趣，希望在您的预训练的基础上重新微调模型。 请问您能否提供在finetune.py中需要load的预训练的模型（如MSBDN/FFANet），以及使用CLAHE对RTTS做预处理的代码呢？ 另外，想确认一下，您使用了CLAHE预处理后的数据进行微调，而不是原始RTTS数据是吗？

zychen-ustc commented 2 years ago

您好。很抱歉由于服务器清理，我目前无法找到pre-train后的模型参数。但是您可以直接利用main.py文件完成MSBDN与FFANet模型的预训练过程(train from scratch)。 关于微调过程中的数据：网络的input仍然是hazy images。由于计算CLAHE loss时需要用到CLAHE处理后的图片，出于加速训练与代码简洁性的考虑，我直接把所有training set里的hazy images处理后，在dataloader里加载。也就是说，dataloader同时加载了hazy image和CLAHE处理后的image，前者为网络输入，后者在计算CLAHE loss时用到。

ZekaiShao25 commented 2 years ago

好的我明白了，谢谢您的及时回复！

ZekaiShao25 commented 2 years ago

还有一个小问题是，您的CLAHE处理的超参是如何选取的？

zychen-ustc commented 2 years ago

我更新了repo，请参考代码CLAHE.py。 另外，如果您是需要CLAHE处理后的RTTS图片，可以直接给我发邮件，可以分享给您。