使用SSD檢測訓練自己的資料

上一篇部落格講到如何制作自己的訓練資料集,這一篇部落格講講如何使用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中模型路徑， 任意找一張圖檔識别，看看效果怎麼樣。