測試darknet 使用voc訓練的模型的標籤問題

使用原網站提供步驟訓練的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命令就可以了。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章