caffe ssd訓(xùn)練自己的數(shù)據(jù)模型

環(huán)境配置完全后

1 VOC數(shù)據(jù)集

在/data/VOCdevkit目錄下創(chuàng)建mydataset, 并放入自己的數(shù)據(jù)集:

cd data/VOCdevkit
mkdir mydataset
cd mydataset
mkdir Annotations
mkdir ImageSets
mkdir JPEGImages
cd ImageSets
mkdir Layout
mkdir Main
mkdir Segmentation

其中Annotations中存放一些列XML文件蔽氨,包含object的bbox,name等;
ImageSets中三個子目錄下均存放train.txt, val.txt, trainval.txt, test.txt這幾個文件厌杜,文件內(nèi)容為圖片的文件名(不帶后綴);
JPEGImages存放所有的圖片渤涌;

2 生成lmdb訓(xùn)練數(shù)據(jù)

在/data目錄下創(chuàng)建一個自己的文件夾:
cd /data
mkdir mydataset

把/data/VOC0712目錄下的create_list.sh 黎比、create_data.sh、labelmap_voc.prototxt 這三個文件拷貝到/mydataset下:
cp VOC0712/create* ./mydataset
cp VOC0712/label* ./mydataset
labelmap_voc.prototxt, 此文件定義label愁憔。

修改create_list.sh
修改成for name in mydataset
修改create_data.sh
修改成dataset_name="mydataset"
修改labelmap_indoor.prototxt
修改分類類別

./data/mydataset/create_list.sh
./data/mydataset/create_data.sh

錯誤處理:
ImportError: No module named caffe.proto
輸入
export PYTHONPATH=/home/cehh/caffe/python:$PYTHONPATH

3 訓(xùn)練

python ./examples/ssd/ssd_mydata2.py
主要修改點(diǎn):
(1)train_data和test_data修改成指向自己的數(shù)據(jù)集LMDB
   train_data = "examples/mydataset/mydataset_trainval_lmdb"
test_data = "examples/mydataset/mydataset_test_lmdb"
(2) num_test_image該變量修改成自己數(shù)據(jù)集中測試數(shù)據(jù)的數(shù)量
(3)num_classes 該變量修改成自己數(shù)據(jù)集中 標(biāo)簽類別數(shù)量數(shù) + 1

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市孽拷,隨后出現(xiàn)的幾起案子吨掌,更是在濱河造成了極大的恐慌,老刑警劉巖脓恕,帶你破解...
    沈念sama閱讀 222,464評論 6 517
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件膜宋,死亡現(xiàn)場離奇詭異,居然都是意外死亡炼幔,警方通過查閱死者的電腦和手機(jī)秋茫,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 95,033評論 3 399
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來乃秀,“玉大人肛著,你說我怎么就攤上這事圆兵。” “怎么了枢贿?”我有些...
    開封第一講書人閱讀 169,078評論 0 362
  • 文/不壞的土叔 我叫張陵殉农,是天一觀的道長。 經(jīng)常有香客問我局荚,道長超凳,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 59,979評論 1 299
  • 正文 為了忘掉前任耀态,我火速辦了婚禮轮傍,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘首装。我一直安慰自己创夜,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 69,001評論 6 398
  • 文/花漫 我一把揭開白布仙逻。 她就那樣靜靜地躺著驰吓,像睡著了一般。 火紅的嫁衣襯著肌膚如雪桨醋。 梳的紋絲不亂的頭發(fā)上棚瘟,一...
    開封第一講書人閱讀 52,584評論 1 312
  • 那天现斋,我揣著相機(jī)與錄音喜最,去河邊找鬼。 笑死庄蹋,一個胖子當(dāng)著我的面吹牛瞬内,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播限书,決...
    沈念sama閱讀 41,085評論 3 422
  • 文/蒼蘭香墨 我猛地睜開眼虫蝶,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了倦西?” 一聲冷哼從身側(cè)響起能真,我...
    開封第一講書人閱讀 40,023評論 0 277
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎扰柠,沒想到半個月后粉铐,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 46,555評論 1 319
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡卤档,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 38,626評論 3 342
  • 正文 我和宋清朗相戀三年蝙泼,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片劝枣。...
    茶點(diǎn)故事閱讀 40,769評論 1 353
  • 序言:一個原本活蹦亂跳的男人離奇死亡汤踏,死狀恐怖织鲸,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情溪胶,我是刑警寧澤搂擦,帶...
    沈念sama閱讀 36,439評論 5 351
  • 正文 年R本政府宣布,位于F島的核電站载荔,受9級特大地震影響盾饮,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜懒熙,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 42,115評論 3 335
  • 文/蒙蒙 一丘损、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧工扎,春花似錦徘钥、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,601評論 0 25
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至橱健,卻和暖如春而钞,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背拘荡。 一陣腳步聲響...
    開封第一講書人閱讀 33,702評論 1 274
  • 我被黑心中介騙來泰國打工臼节, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人珊皿。 一個月前我還...
    沈念sama閱讀 49,191評論 3 378
  • 正文 我出身青樓网缝,卻偏偏與公主長得像,于是被迫代替她去往敵國和親蟋定。 傳聞我的和親對象是個殘疾皇子粉臊,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,781評論 2 361

推薦閱讀更多精彩內(nèi)容