windows 10+ubuntu16.04全程配置CPU版本的caffe ssd並訓練自己的數據研究報告

本人花了好幾天,順利完成windows版本caffe-ssd和ubuntu版本的caffe-ssd環境配置、腳本修改、最終獨創一套高速自定義訓練的項目包,可以在配置caffe ssd環境後可以迅速訓練自己的數據,而不需要眼花繚亂的搞各種腳本,進行各種轉換,這麼麻煩了。雖然談不上是專家,但是已經比較熟了,裏面很多腳本都是本人親自操刀書寫。先來一張windows下面訓練自己定義的數據窗口顯示

再來一張ubuntu下面CPU版本訓練的caffe ssd截圖

兩個平臺的都差不多但是腳本有點區別,windows主要使用bat文件,而ubunutu主要是用python和shell腳本,使用方法我分爲2個平臺介紹,在windows下面,只要放到指定位置,基本不用改代碼,這是windows上面好處

實驗環境:

Windows10專業版64位

Caffe-ssd CPU版本

實驗要求:

源碼必須放在D:\caffe-ssd-microsoft,然後將myproj放進caffe-ssd-microsoft文件夾

 

 

(1)打開myproj文件夾進去data文件夾裏面的

D:\caffe-ssd-microsoft\myproj\data\VOCdevkit\VOC2012然後JPEGImages放進去自己的jpg文件,Annotations放進自己的標註文件

  1. 刪除D:\caffe-ssd-microsoft\myproj/lmdb裏面的train_lmdb文件夾和test_lmdb文件夾,不然生成lmdb會提示錯誤
  2. bat文件夾裏面有個xml2txt.py,這個是將圖片轉爲txt文件,dos窗口切換到bat目錄,執行python xml2txt.py,這時候txt文件夾會生成對應的train.txt和test.txt,裏面的內容是圖片文件名 xml文件名這樣的格式,注意這個腳本是將數據集80%作爲訓練集,20爲測試集,可以再腳本里面改
  3. bat文件夾裏面有個get_name_size.py,dos窗口切換到bat目錄,執行python get_name_size.py,這時候txt文件夾會生成對應的test_name_size.txt,裏面的內容是文件名無後綴 高 寬這樣的格式
  4. 雙擊bat文件夾裏面create_lmdb.bat即可生成train_lmdb和test_lmdb,注意雙擊前確保train_lmdb和test_lmdb文件夾不要存在,否則會報錯
  5. 雙擊bat文件夾裏面的starttrain.bat即可開始訓練、

在ubunutu平臺,由於都是使用的絕對路徑需要改自己的用戶名,其他基本不用改

實驗環境:

Ubuntu16.04

Caffe-ssd CPU版本

實驗要求:

新建一個/home/caffe-ssd目錄,將caffe放進去,然後將myproj文件夾和caffe放同級目錄

 

 

(1)打開myproj文件夾進去data文件夾裏面的

/home/fut/caffe-ssd/myproj/data/VOCdevkit/VOC2012

然後JPEGImages放進去自己的jpg文件,Annotations放進自己的標註文件

  1. 刪除/home/fut/caffe-ssd/myproj/lmdb裏面的train_lmdb文件夾和test_lmdb文件夾,不然生成lmdb會提示錯誤
  2. 打開xml2txt.py修改裏面用戶名,我的是fut改改用戶名就可以了,用替換的方法就可以,然後終端切換到script目錄,執行python xml2txt.py,這時候txt文件夾會生成對應的train.txt和test.txt,裏面的內容是圖片文件名 xml文件名這樣的格式,注意這個腳本是將數據集80%作爲訓練集,20爲測試集,可以再腳本里面改。如果你遇到caffe ssd安裝配置問題,可以私信我,有償幫忙。
  3. 打開get_name_size.py修改裏面用戶名,我的是fut改改用戶名就可以了,用替換的方法就可以,然後終端切換到script目錄,執行python get_name_size.py,這時候txt文件夾會生成對應的test_name_size.txt,裏面的內容是文件名無後綴 高 寬這樣的格式
  4. 終端切換到script目錄,執行python train_lmdb_gen.py即可生成lmdb,生成的文件在lmdb裏面查看
  5. 終端切換到script目錄,執行python test_lmdb_gen.py即可生成lmdb,生成的文件在lmdb裏面查看
  6. 終端切換到script目錄,執行python ssd_pascal.py即可開始訓練,其中很多參數ssd_pascal.py可以修改,這裏只需要把用戶名換成自己的就可以了,這個是做檢測用的。

 

由於手頭沒有gpu電腦所以沒有配置GPU版本,但是這些都是小CASE。

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