使用原網站提供步驟訓練的VOC數據集的方法,最終可以得到yolov3-416的模型。
但需要使用darknet detector test <cfg/voc.data> <cfg/yolov3-voc.cfg> <weights> <your image> 命令進行測試
但如果使用簡潔的測試命令會發現檢測的標籤名稱不正常。例如
$sudo ./darknet detect cfg/yolov3-voc.cfg <weights> data/dog.jpg
.........
data/dog.jpg: Predicted in 0.016741 seconds.
stop sign: 85%
train: 100%
bicycle: 99%
正常的結果應該是
.......
data/dog.jpg: Predicted in 0.016686 seconds.
dog: 85%
car: 100%
bicycle: 99%
原因是darknet 在使用detect 參數時,指定了 "cfg/coco.data"。
如果有強迫症,可以有兩個方法解決:
1.修改源碼examples/darknet.c,增加一套參數,調用cfg/voc.data,然後重新編譯。
2.替換cfg/coco.data內容爲cfg/voc.data的內容。(這樣名字和內容不一致,心裏要知道)
當然模型本身是正常的,使用darknet detector test命令就可以了。