YOLOV5:数据集制作

YOLOV5:数据集制作

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

YOLOV5:数据集制作_第1张图片

第一步:使用labellmg进行标记

YOLOV5:数据集制作_第2张图片

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

YOLOV5:数据集制作_第3张图片

split.py文件配置

YOLOV5:数据集制作_第4张图片

--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

YOLOV5:数据集制作_第5张图片

YOLOV5:数据集制作_第6张图片

YOLOV5:数据集制作_第7张图片

第四步:random1.py

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

注意,这里不能命名random,因为python中有random函数,pycharm分不清是调函数还是调程序
YOLOV5:数据集制作_第8张图片

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

YOLOV5:数据集制作_第9张图片

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

YOLOV5:数据集制作_第10张图片
YOLOV5:数据集制作_第11张图片

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

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

你可能感兴趣的:(YOLOV5,Deep,Learning,python,python,pytorch,tensorflow)