yolo-v2 自己的数据集训练流程(仅供内部使用!)

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接: https://blog.csdn.net/Dontla/article/details/102745998

warning

该流程仅供内部使用,外部人士使用可能会报很多很多错误!

步骤

  1. 先清除backup文件夹中老的权重文件:
    在这里插入图片描述
  2. 将标定好图片以及annotation .txt文件拷贝到obj文件夹,一一对应,路径build/darknet/x64/data/obj,如:
    在这里插入图片描述
  3. 修改train.txt和test.txt文件,路径需与我们的图片以及标注路径一致:
    在这里插入图片描述
    详细步骤:
    1)新建Exel表格,在第一个单元格输入第一张图片路径,按住单元格右下角按钮往下拖动,直到生成跟数据集相同数量的单元格:
    在这里插入图片描述
    2)ctrl+a全选单元格,复制,打开train.txt文件,粘贴:
    在这里插入图片描述
    3)自行从中抽取(剪切)若干行粘贴到test.txt中,作为测试集,我这里抽取301-400、701-800、1101-1200、1501-1600共400张(25%)作为测试集。如有条件可用数据集拆分工具,更精准:
    在这里插入图片描述
  4. 修改obj.data和obj.names文件,路径要对应:
    在这里插入图片描述
    在这里插入图片描述
  5. 复制yolo-voc.2.0.cfg文件,重命名为yolo-obj.cfg,打开yolo-obj.cfg,修改里面相关参数(属于高级调参,这步我们暂时不修改):
    在这里插入图片描述
  6. 打开darknet.sln,修改darknet.c相关代码,点击生成–> 生成解决方案:
    在这里插入图片描述
    在这里插入图片描述
    在这里插入图片描述
  7. 成功生成解决方案后,用记事本打开train-obj.cmd,查看里面内容,确认无误后,运行它,便开始训练,训练过程中可查看backup文件夹生成的权重文件情况:
    在这里插入图片描述
    在这里插入图片描述
    在这里插入图片描述
    在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/Dontla/article/details/102745998
今日推荐