tiny-yolo 訓練自己的數據集

感謝團隊裏面小夥伴們。

tiny-yolo 使用自己的數據集進行訓練

./darknet partial cfg/darknet.cfg darknet.weights darknet.conv.10 10

./darknet detector train cfg/voc.data cfg/tiny-yolo-voc.cfg darknet.weights

下載

1.darknet.cfg https://pjreddie.com/darknet/imagenet/#extraction

點擊cfg file
選擇原因：較小的模型，適用於tiny-yolo的速度快

運行

./darknet partial cfg/darknet.cfg darknet.weights darknet.conv.10 10

選擇10的原因：tiny-yolo 卷積層數爲9，yolo 卷積層數爲22（參考得出）

訓練

./darknet detector train cfg/voc.data cfg/tiny-yolo-voc.cfg darknet.weights

採用剛剛生成的預訓練模型進行訓練