毕设日志——2019年1月14日

版权声明:本文为博主原创文章,未经博主允许不得转载。 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版本代码。

感想

做事情要有条理,磨刀不误砍柴工!

保证效率和自己的步伐,不被他人进度左右!

猜你喜欢

转载自blog.csdn.net/qq_27022241/article/details/86476203