我看了网上很多yolo教程,可能是因为电脑环境和配置的不一样,所以我并没有完全通过网上教程成功训练自己的数据集,接下来我将我自己完全亲自测试并且最后成功训练数据集的教程记录如下:
训练自己的数据集有如下几步:
-
准备数据集并且安装yolo
-
制作数据集的标签生成xml文件
-
将xml文件转成txt文件
-
修改配置文件
-
下载预训练模型进行训练
**第一步:**安装yolo并且准备自己的数据集(图片格式是jpg格式),安装yolo参照网址:https://pjreddie.com/darknet/yolo/
**第二步:**制作数据集标签生成xml文件:
大家在这里需要安装安装 labelImg (标数据用),我在网上试了很多教程,最后我推荐大家按照这个网址的教程装:https://www.cnblogs.com/new-age/p/7071289.html 装完之后用labelImg做标签并将生成的xml文件保存在xml文件夹下。