1. 程式人生 > >貓狗大戰-caffe模型訓練例項&NSDK識別執行

貓狗大戰-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.