修改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:測試集圖檔數目