windows下使用Caffe框架和matlab实现SRCNN官方代码的步骤
步驟
step1 搭建caffe環境
在windows系統上搭建caffe環境,并配置matlab接口(需要下載 caffe-master.zip 以及 VS2013 )
我的環境為:windows10 + caffe-master + VS2013旗艦版 + matlab R2018b
安裝步驟推薦參考:
博客1地址:https://blog.csdn.net/jiangjieqazwsx/article/details/53292326
Win10 + Caffe + CPU + MATLAB (包括各種問題詳細解決)(一) —— VS2013 端設置
Win10 + Caffe + CPU + MATLAB (包括各種問題詳細解決)(二) —— MATLAB 端設置
博客2地址:
https://blog.csdn.net/u014546828/article/details/80447583
https://blog.csdn.net/u014546828/article/details/80431134
win7 64位+caffe+cuda7.5配置
博客3地址:https://blog.csdn.net/XZZPPP/article/details/51510785
[注意] 在使用VS生成解決方案過程中,可能會報很多錯誤,需要你耐心去解決,別的博客好像都只報過一兩個錯,但是我是解決一個錯誤又一個錯誤,得有七八個,最后還是配好了
step2 訓練SRCNN—>SRCNN_train
(1). 解壓 SRCNN_train 文件
- 將 SRCNN_train 的 SRCNN文件夾放到 /caffe-master/examples下面,例如 D:\Program Files\caffe-master\examples\SRCNN
(2).生成訓練和測試用的.h5文件
- 運行其中的 generate_train.m,generate_test.m 生成train.h5、test.h5文件, caffe可對 HDF5 數據格式進行操作,Caffe數據類型參考 http://www.cnblogs.com/denny402/p/5073427.html
[注意] train.h5 和 test.h5 文件名稱要固定,這兩個文件名與train.txt、test.txt、SRCNN_solver.prototxt 文件中的定義相一致,擅自更改易出錯,這個test.h5 數據文件也是訓練時用的,與我們測試效果所用的測試數據概念上進行區分
(3). 生成.bat文件并運行
-
在 caffe-master 下面新建一個 .bat 文件(windows批處理文件,可自定義名稱,但后綴一定要是.bat),文本內容為 .\Build\x64\Release\caffe.exe train --solver examples/SRCNN/SRCNN_solver.prototxt
-
右鍵,以管理員方式運行 bat 文件,就開始進行SRCNN訓練了
根據SRCNN_solver.prototxt的設置,訓練數據每500次迭代會進行一次測試,并生成.caffemodel 和 .solverstate文件,該文件中還有學習率和動量參數等的設置,自己打開看進行理解吧
(4).運行saveFilters.m
- 運行saveFilters.m,生成 mat 格式的卷積核訓練文件。
step3 測試->SRCNN_v1
- 測試是使用 SRCNN_v1 進行測試,先對其進行解壓,然后運行 SRCNN_v1 里面的 demo_SR 即可
其他改進模型
網址:https://blog.csdn.net/qq_26499769/article/details/79628851
除了上述列出的博客,還參考了以下博客:
在windows的Caffe框架下實現SRCNN
bvlc_reference_caffenet.caffemodel 下載鏈接
總結
以上是生活随笔為你收集整理的windows下使用Caffe框架和matlab实现SRCNN官方代码的步骤的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: Bootstrap模态框显示时有阴影遮罩
- 下一篇: linux 修改ldap密码,Linux