天天看點

caffe 訓練自己的資料集

修改ssd_pascal.py即可。

82行:train_data路徑; 
84行:test_data路徑; 
86行:resize_width,輸入網絡的圖檔寬度
87行:resize_height,輸入網絡的圖檔高度
232行:base_lr,設定基礎學習率,實際值是base_lr×25
237-246行:model_name、save_dir、snapshot_dir、job_dir、output_result_dir路徑; 
259-263行:name_size_file、label_map_file路徑; 
266行:num_classes修改為1 + 類别數;
299行:min_dim設定為resize_width或resize_height中的最小值,如果沒有設定過resize_width或者resize_height,則不需要設定。 
360行:num_test_image:測試集圖檔數目