windows下的caffe实现SRCNN

本文转载自https://blog.csdn.net/whr_ws/article/details/82823271

文字简单实用,上面这个也是转载的,最初的作者找不到了,感谢分享!

本次实验是实现SRCNN。windows下的caffe框架为caffe-master.zip,直接解压即可,要用到caffe提供的matlab接口,参考https://blog.csdn.net/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.csdn.net/qq_26499769/article/details/79628851

在参考这篇文章之后又遇到了一些问题,在另一篇博客总结出来https://mp.csdn.net/postedit/85224812

猜你喜欢

转载自blog.csdn.net/qk61508/article/details/85224326
今日推荐