YOLOV5:数据集制作

2021/9/21 6:28:51

本文主要是介绍YOLOV5:数据集制作,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

YOLOV5:数据集制作

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

请添加图片描述

第一步:使用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

请添加图片描述

请添加图片描述

请添加图片描述

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

请添加图片描述

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

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

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

nc: 2
names: ['锡纸', '塑料']


这篇关于YOLOV5:数据集制作的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程