图像超分辨率重建之srcnn,基于tensorflow实现
Posted zzydaydayup
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了图像超分辨率重建之srcnn,基于tensorflow实现相关的知识,希望对你有一定的参考价值。
本篇适用人群对于那些知道srcnn的每个步骤的人但是不是很会打代码的人
首先,附上我的github:https://github.com/zzydashuaibi/srcnn_tensorflow
在写代码之前,我们需要明白一件事就是我们每一次训练实际上是训练图片的某一部分(33*33)最后输出的是卷积后的大小只有22*22,所以srcnn的预处理要比其他的图像重建的模型要复杂一点。
他除了一般的预处理操作,还需要将图片分割,最后的训练完还做实验的时候还需要将图片结合起来,至于其他的,相信你自己,虽然第一次可能会觉得难,但是多看几天,总会看的懂的。
那么一点点注释:config.py是用来配置一些训练参数,以及sample,chekpoint的路径
evaluate.py是用来做评价函数的,但是这次我没有用它,也就是ssim与psnr
model.py是模型文件以类的方式实现
preprocessing.py是预处理文件
test.py是那张图来测试这个模型
同时,这个代码逻辑是很多人不曾谈的,导致下个代码却读起来很难受,因此我在代码中注解了很多注释,同时也将很多东西用最简单的方式写出。
最后,refference:tegg89的这篇代码
以上是关于图像超分辨率重建之srcnn,基于tensorflow实现的主要内容,如果未能解决你的问题,请参考以下文章
图像重建基于matlab卷积神经网络的图像超分辨率重建含Matlab源码 1816期