JavaShuo
Windows下使用yoloV3训练自己的数据
时间 2021-01-02
标签
深度学习#反向传播算法
配置好yolov3后就可以训练自己的数据
进行数据标注
,网上很多人是用labelimg进行标注的,标注之后还要进行文件的转换,这里我直接使用yolo提供的标注工具进行标注
yolo_mark的使用可以参考这个博客
下载地址:
https://github.com/AlexeyAB/Yolo_mark
https://blog.csdn.net/heiheiya/article/details/83506913
根据自己的数据设定类别,进行标注即可
标注之后开始训练
在D:\Practice Program\DarkNet\darknet-master目录下创建一个新的文件夹,用来存放刚刚标注完的文件,这里我新建了一个batteryDataset,文件夹里包括训练集,测试集数据以及一些配置文件
接着进入到cfg文件夹下,复制一份yolov3-.cfg文件,将其重命名,之后打开对其进行修改:
可以根据自己的电脑配置进行设置,
我的显卡不支持多尺度训练,因此
random要改为1
开始训练
下载
打开cmd,切换至darknet-master目录下,输入darknet.exe detector train [你的.data文件存放的相对路径] [你的.cfg文件存放的相对路径] 上面下载文件夹存放的路径 因为使用相对路径最好都存放在darknet-master目录下,多建几个文件夹即可