本次实验是实现SRCNN。windows下的caffe框架为caffe-master.zip,直接解压即可,要用到caffe提供的matlab接口,参考https://blog.youkuaiyun.com/jiangjieqazwsx/article/details/53292326
(提示matlab的版本选2013及其以上版本的,matlab2013以下版本缺少GPU的头文件)安装时注意环境变量的设置。
解压SRCNN_train文件,,将SRCNN文件夹放到caffe-master/examples下面,例如E:\caffe-master\examples\SRCNN
接着运行其中的generate_train.m,generate_test.m生成train.h5、test.h5文件, caffe可对HDF5数据格式进行操作,Caffe数据类型参考http://www.cnblogs.com/denny402/p/5073427.html
在caffe-master下面新建一个bat格式的文本,文本内容为
.\Build\x64\Release\caffe.exe train --solver examples/SRCNN/SRCNN_solver.prototxt
pause
然后运行bat文本就能进行CNN训练,生成caffemodel的文件,接着运行saveFilters.m,生成mat格式的卷积核训练文件。
最后运行SRCNN_v1里面的demo_SR即可
其他改进模型可在网上搜到,网址https://blog.youkuaiyun.com/qq_26499769/article/details/79628851
本文详细介绍了如何在Windows环境下使用Caffe框架实现SRCNN超分辨率卷积神经网络,包括caffe-master的解压配置、MATLAB接口安装、环境变量设置、SRCNN文件夹放置、数据文件生成、CNN训练及卷积核训练文件生成步骤。
1725





