在windows的Caffe框架下实现SRCNN

本文详细介绍了如何在Windows环境下使用Caffe框架实现SRCNN超分辨率卷积神经网络,包括caffe-master的解压配置、MATLAB接口安装、环境变量设置、SRCNN文件夹放置、数据文件生成、CNN训练及卷积核训练文件生成步骤。

本次实验是实现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

评论 11
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值