NO IMAGE

上一篇部落格講到如何製作自己的訓練資料集,這一篇部落格講講如何使用SSD訓練自己的資料.

在訓練資料做好後。訓練程式為/examples/ssd/ssd_pascal.py,執行之前,我們需要修改相關路徑程式碼,主要是訓練路徑的修改和關於自己資料集引數的一些修改.

cd /examples/ssd
先複製一份sd_pascal.py, 再開啟,把裡面的路徑修改為自己的.具體有如下:
 train_data路徑;
test_data路徑;
save_dir、snapshot_dir、job_dir、output_result_dir路徑;沒有的話就仿照voc0712的資料夾進行建立.
name_size_file、label_map_file路徑;

name_size_file

label_map_file
num_classes 修改為1 類別數
num_test_image:測試集圖片數目

batch_size看自己電腦效能進行修改吧,

另外, 如果你只有一個GPU, 需要修改285行: 
gpus=”0,1,2,3” ===> 改為”0” 
否則,訓練的時候會出錯。 

另外,自己的電腦沒有GPU的話,最好不要訓練,本人一訓練就宕機.尷尬.

修改完後執行

python ./examples/ssd/ssd_pascal.py 

訓練完, 修改ssd_detector.py中模型路徑, 任意找一張圖片識別,看看效果怎麼樣。