search

hellloxiaotian / ADNet

Attention-guided CNN for image denoising(Neural Networks,2020)
https://doi.org/10.1016/j.neunet.2019.12.024
117 stars 5 forks source link

对其他噪声的去噪效果 #29

Open 6MrZou6 opened 2 years ago

6MrZou6 commented 2 years ago

田教授您好！我按照您所说的重新写了一个生成椒盐噪声的代码，并重新生成训练和测试的h5文件，将pristine_color_image添加椒盐噪声训练了两个epoch，训练的效果都很好。用得到的model_1和model_2对含椒盐噪声的图片进行测试，测试结果发现对椒盐噪声去噪几乎不起作用，请问这是因为epoch少了的原因吗？因为训练的确很花时间。

hellloxiaotian commented 2 years ago

你好： 感谢关注。主要的原因可能是Epoch少了，可以训练久一些再看看；你还可以再增加一些数据量和使用一些数据增强的方法，等模型快收敛了再看看最后的效果。

6MrZou6 commented 2 years ago

谢谢教授，请问您当初对高斯噪声去噪训练70个epoch具体达到的psnr是多少呢？我目前训练到4个epoch psnr才35左右 且效果还不好