版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_27022241/article/details/86476203
毕设日志——2019年1月14日
(2018.1.9~2018.1.14总结)
一、DnCNN PyTorch代码
下载(https://github.com/SaoYan/DnCNN-PyTorch)并按照知乎博客PyTorch实战指南(https://zhuanlan.zhihu.com/p/29024978)规范化DnCNN训练去噪模型代码。
二、进一步整理代码,使之适合训练和测试DnCNN-S和DnCNN-B模型
1. 增加评价指标SSIM(多尺度SSIM)
参考:https://github.com/jorge-pessoa/pytorch-msssim
2. visdom可视化
(1)loss曲线
(2)PSNR和SSIM曲线
3. 保存模型
4. 保存日志信息
(1)控制台信息
(2)默认参数设置信息
5. 运行时间显示
time.clock()
三、问题遗留
未找到FSIM的PyTorch版本代码。
感想
做事情要有条理,磨刀不误砍柴工!
保证效率和自己的步伐,不被他人进度左右!