tiny-yolo训练测试(coco训练集混合自己的voc训练集)--20170829

一、构建VOC数据集

1.准备数据
(1)./write.sh //设置输入输出文件的路径,包含训练集图片的文件名
注:输出文件路径是train.txt的绝对路径,并去掉所以后缀名.xml。
(2)运行voc_label.py,生成train_2017.txt文件,里面包含了所有训练样本的绝对路径。
(3)把coco数据集(jason)转换成voc数据集同样的格式,并将coco和自己的数据集混合成新的数据集。
运行darknet-master/data_coco/coco-master/PythonAPI/pycocotools/coco-label.py

二、修改配置文件:

1、修改darknet-master/data/anngic.names里面的类别。
2、修改cfg/coco.data 里面的类别、权重保存文件夹等。
3、修改cfg/tiny_yolo_coco.cfg里的类别、最后一层卷积层计算次数,
修改anchors:
注:像素416*416时用第一行;像素288*288时用第二行。
修改训练次数,如:max_batches=200000
注:接着之前的训练结果训练就把次数累加,不用就直接写训练次数。

三、训练

3.1 训练新的模型,生成只有前13层的权重文件,后两层(14、15层)和分类有关。

#只保留tiny-yolo-coco_final.weights模型的前13层,后两层和分类有关,新模型保存到pre-trained_weights/tiny-yolo-coco.conv.13。

./darknet partial cfg/tiny-yolo-coco.cfg backup/tiny-yolo-coco_288-2.0/tiny-yolo-coco_final.weights  pre-trained_weights/tiny-yolo-coco.conv.13 13

                
  • 1
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值