一、環境介紹
主機系統ubuntu14.04,使用caffe平臺,使用的python2.7
二、準備工作
1.準備自己的數據集(包括訓練集、測試集)我的包含如下內容,數據集製作目前還不瞭解,就不介紹了。
存儲在下載的FCN源碼位置:/home/qiu/code/fcn.berkeleyvision.org-master/data/pascal/VOC2007。這是我的路徑。
2.下載FCN源碼,fcn源碼在github裏面下載即可。
三、更改文件
我們使用voc-fcn32s來訓練我們的數據集。先看最終文件夾包含的文件,沒有的文件都在fcn.brek....源碼下拷貝進來就行。
1.voc-fcn32s/solve.py更改三個地方
2.train.prototxt
3.val.protptxt
4.voc_layers.py
50行改成這樣
100行改成這樣
113行改成這樣
215行改成這樣
223行往後改成這樣
5.新建deploy.prototxt文件
建好這個文件,並命名後,將train.prototxt文件的內容全部拷貝過來。然後將train的第一層改爲
在者就是刪除最後一層layer就行了。上面訓練的部分就改完了。
6.這個是源碼文件下的infer.py的更改
17行改成這樣
這個文件改完是可以利用voc_fcn32s測試的。