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

采用刚刚生成的预训练模型进行训练