NAFNet
NAFNet copied to clipboard
请问文中的baseline开源了吗
好像代码里只有nafnet没有baseline
另外还有一个问题,我尝试了在自己的deblur数据集上训练nafnet,发现收敛的非常慢,需要几千epoch才能达到理想的性能。 可以请问下作者这是为什么吗?感谢解答
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)。