NAFNet icon indicating copy to clipboard operation
NAFNet copied to clipboard

请问文中的baseline开源了吗

Open ciki000 opened this issue 2 years ago • 2 comments

好像代码里只有nafnet没有baseline

ciki000 avatar Jul 29 '22 10:07 ciki000

另外还有一个问题,我尝试了在自己的deblur数据集上训练nafnet,发现收敛的非常慢,需要几千epoch才能达到理想的性能。 可以请问下作者这是为什么吗?感谢解答

ciki000 avatar Jul 29 '22 10:07 ciki000

ciki000, 您好, Baseline现已开源, 欢迎试用。

对于收敛的比较慢的问题, 我无法下定论, 根据以往的经验和模型规模有一定关系(相同时间下, 更大的模型取得的收敛效果比较好)。 和之前的方法比的话, 在同样模型规模下(16GMACs), HINet (Simple)在REDS下, 300k, 512x512, batch_size 32, PSNR 28.23; 而NAFNet, 200k, 256x256, batch_size 32, PSNR 达到28.84 (相当于训练量仅HINet Simple 的1/6)。

mayorx avatar Aug 03 '22 03:08 mayorx