Skip to content

Latest commit

 

History

History
28 lines (21 loc) · 561 Bytes

File metadata and controls

28 lines (21 loc) · 561 Bytes

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