貓狗大戰-caffe模型訓練例項&NSDK識別執行
首先【caffe-Windows】安裝,請參考https://blog.csdn.net/zb1165048017/article/details/51355143(博主寫的非常詳細,避免了很多坑)
注:如果出現libcaffe.lib無法開啟的錯誤且按照博主寫的策略無法解決,請重新啟動vs2013,其餘完全按照博主步驟操作,博主沒有一點點多餘的描述,全部是重點,切記仔細認真的按照步驟操作。我是一個不安分的散人,不按照博主的思路自己去摸索走了很多彎路,中間自己挖了很多坑,浪費了不少時間,參考文件還可以看下https://www.cnblogs.com/yixuan-xu/p/5858595.html我自己的電腦沒有GPU,只能用CPU訓練,一天一夜整整24小時訓練結束。
首先生成自己的graph
mvNCCheck -w stored_model_iter_10000.caffemodel -i ../../data/images/cat.jpg -s 12 dogsvscats.prototxt -S 255
mvNCCompile dogsvscats.prototxt -w stored_model_iter_10000.caffemodel -s 12 -is 208 208 -o graph
引數含義略過
參考文獻
自己挖的坑,以及自己跳出來:
坑0. 配置環境的時候由於電腦有VS2017,就沒安裝VS2013,結果各種調不通
跳坑:安裝VS2013,由於網速慢,足足耗了一天。
坑1. 配置環境出現python不支援,無奈註釋了python.
跳坑:修改CommonSettings.props配置檔案,把python那一項改為false。
坑2. 生成檔案的時候,出現libcaffe.lib無法開啟的錯誤並且按照作者方法無法通過。
跳坑:重新開啟VS2013
坑3. 訓練貓狗大戰模型出現各種系統崩潰,報錯。
跳坑:回頭看作者文章,捋第一遍,發現驗證集沒準備(犯懶的毛病)解決一個報錯,再次執行還是有錯;
捋第二遍,發現驗證集list沒有(已經抓狂了),老老實實把前提都按照作者的要求做了,發現:lmdb資料生成的有問題,模型檔案中文沒去掉,正如作者所寫,都是因為不仔細犯的錯,從頭到尾再來了一次。終於訓練開始了。
坑4. 分類預測出現錯誤,無法識別data,各種找資料,搜帖子,無解。再次捋一遍文章,發現沒有理解prototxt檔案的意思,沒有徹底理解網路的含義,沒有仔細按照要求更改prototxt,
跳坑:按照要求修改prototxt,執行test.bat.