ubuntu下FCN跑自己的訓練集、測試集

一、環境介紹

主機系統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測試的。

 

 

 

 

 

 

 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章