Java教程

YOLOV5：数据集制作

本文主要是介绍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：数据集制作的文章就介绍到这儿，希望我们推荐的文章对大家有所帮助，也希望大家多多支持为之网！
您可能喜欢
栏目导航