Windows下使用yoloV3训练自己的数据

配置好yolov3后就可以训练自己的数据

  1. 进行数据标注,网上很多人是用labelimg进行标注的,标注之后还要进行文件的转换,这里我直接使用yolo提供的标注工具进行标注
    yolo_mark的使用可以参考这个博客
    下载地址:https://github.com/AlexeyAB/Yolo_mark
    https://blog.csdn.net/heiheiya/article/details/83506913
    根据自己的数据设定类别,进行标注即可
  2. 标注之后开始训练
    在D:\Practice Program\DarkNet\darknet-master目录下创建一个新的文件夹,用来存放刚刚标注完的文件,这里我新建了一个batteryDataset,文件夹里包括训练集,测试集数据以及一些配置文件
    在这里插入图片描述
    接着进入到cfg文件夹下,复制一份yolov3-.cfg文件,将其重命名,之后打开对其进行修改:
    在这里插入图片描述
    可以根据自己的电脑配置进行设置,
    我的显卡不支持多尺度训练,因此
    在这里插入图片描述
    random要改为1

开始训练

  1. 下载
    在这里插入图片描述
  2. 打开cmd,切换至darknet-master目录下,输入darknet.exe detector train [你的.data文件存放的相对路径] [你的.cfg文件存放的相对路径] 上面下载文件夹存放的路径 因为使用相对路径最好都存放在darknet-master目录下,多建几个文件夹即可