README.md

AI cup

Download frozen_inference_graph.pb

link 放到 out_graph_dir/saved_model_6000steps/

CPU GPU設定

在 detect.py 第43行

• 有GPU

os.environ['CUDA_VISIBLE_DEVICES'] = '0'

• 沒GPU

os.environ['CUDA_VISIBLE_DEVICES'] = '-1'

執行 detect

-i 為要偵測的圖片資料夾, -o 為要輸出的圖片資料夾

python detect.py -i Dev/ -o Result/

Result