使用訓練好的YOLO進行檢測

• 首先安裝Darknet

git clone https://github.com/pjreddie/darknet
cd darknet
make

• 下載預訓練得到的權重檔案

YOLO的配置檔案在./cfg/資料夾下。現在下載預訓練網路的權重：

wget https://pjreddie.com/media/files/yolo.weights

• 執行檢測器

./darknet detect cfg/yolo.cfg yolo.weights data/dog.jpg

這個檢測命令等價於

./darknet detector test cfg/coco.data
 
 cfg/yolo.cfg yolo.weights data/dog.jpg

前者是後者的簡寫形式。

• 結果

執行檢測命令後，會在螢幕上列印下面的資訊：

這裡預設是對./data/dog.jpg圖片進行檢測，生成檢測結果圖片為./predictions.png，如下圖所示：

• 對多張圖片進行檢測

如果想對多張圖片進行檢測，輸入如下命令：

./darknet detect cfg/yolo.cfg yolo.weights

這樣會出現如下的提示：

輸入待檢測圖片的檔名，比如data/person.jpg（一次只能輸入一個檔名），這樣在檢測完這張圖片之後會再次出現此提示，接著輸入下一張待檢測圖片檔名。CTRL+C可以退出。

參考

[1] 作者實現