将HSDT在HSIR上使用pretrained model测试可以达到论文中的PSNR，但使用提供的训练集训练无法达到论文提到的PSNR（ch1263152934@163.com）

Zeqiang-Lai / HSDT

Hybrid Spectral Denoising Transformer with Guided Attention (ICCV 2023)

http://arxiv.org/abs/2303.09040

MIT License

41 stars 4 forks source link

Open 1263152934 opened 8 months ago

1263152934 commented 8 months ago

除了xavier_normal_的权重初始化，training是否还有其他细节，如果方便的话能否提供原始的训练代码

Zeqiang-Lai commented 7 months ago

原始代码已经上传了，可以试试，hsir 那套重构的代码我之前复现确实有点问题（不过我自己对过好几遍也没找出原因在哪），原始代码训过很多次应该没有问题，如果有问题可以再讨论。

1263152934 commented 7 months ago

好的，非常感谢，麻烦您了