YOLOV5:数据集制作【照着做你也能训练自己的神经网络】

7 篇文章 6 订阅
5 篇文章 19 订阅

YOLOV5:数据集制作

总体流程:
labellmg标记(.xml)-> split.py (生成四个集的txt)-> txt2yolo.py -> 各个图像的txt -> 按照标准规范文件夹 -> 制作yaml文件

请添加图片描述

(这里注意,要把照片全部换成jpg格式,通过函数转换,而且照片名字中不能带中文)

第一步:使用labellmg进行标记

请添加图片描述

快捷键:请添加图片描述

在汇总的时候要检测是不是一张照片对应一张标签:
这里可以设置为中图标,一个照片一个标签,看哪里出问题了。
请添加图片描述

第二步:用split.py在新文件中生成四个txt,同时在四个txt中分配好了哪些图片是训练集哪些是验证集

请添加图片描述

split.py文件配置

请添加图片描述

--xml
C:\Users\zhaohaobing\PycharmProjects\python-yolov5-project\Data_set_processing\imagesplus
--txt
C:\Users\zhaohaobing\PycharmProjects\python-yolov5-project\Data_set_processing\imagespluss

第三步:xml2yolo.py

请添加图片描述

请添加图片描述

请添加图片描述

这里会报的错:
是照片和标签名字中有中文!!!
请添加图片描述
解决办法:
第一种:
根据报错的文件名字去.xml和.jpg文件夹搜
请添加图片描述

第二种:
在生成的图片标签txt文件的文件夹种看卡在哪个文件了,再回去找.xml文件中
请添加图片描述

第四步:random1.py

随机将照片和标签分配到train和valid中

注意,这里不能命名random,因为python中有random函数,pycharm分不清是调函数还是调程序
请添加图片描述

第五步:标准化数据集文件夹格式

请添加图片描述

第六步:制作配置文件 data.yaml

请添加图片描述
请添加图片描述

配置文件里要写:数据都去哪读,nc类别数目
train: ../train/images
val: ../valid/images

nc: 2
names: ['锡纸', '塑料']
  • 23
    点赞
  • 125
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 9
    评论
评论 9
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

hbzhao@USTC

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值