（4）YOLOV5算法二之数据集转换及自动划分、随机划分训练集与测试集

明月醉窗台

已于 2023-04-18 18:13:17 修改

阅读量744

点赞数 2

分类专栏： # 运动目标检测人工智能文章标签：算法计算机视觉 opencv 人工智能目标检测

于 2022-08-05 16:05:11 首次发布

本文链接：https://blog.csdn.net/yohnyang/article/details/126179584

版权

人工智能同时被 2 个专栏收录

33 篇文章 32 订阅 ¥159.90 ¥299.90

订阅专栏

运动目标检测

25 篇文章 25 订阅

订阅专栏

本文介绍了如何进行YOLOV5的数据集转换，包括将图像转为.jpg格式，标签.xml转为.txt格式，并详细阐述了数据集的自动划分过程，通过设置比例创建训练集、测试集和验证集。

摘要由CSDN通过智能技术生成

1.数据转换

使用背景：多个文件夹及标签.xml文件(labelImg标注)
实现思路：
1.将所有的原始数据.jpg .png .xml全放到一个文件夹
2.图像均转为.jpg格式，放入图像文件夹
3.标签由.xml转为yolov5需要的格式.txt

具体实现：

"""
time:20220729
writer:yohn
function:制作自己的原始数据集，转换成yolo需要的格式
"""

import os
import shutil
import cv2
import xml.etree.ElementTree as ET

#待处理数据文件夹，将图像与标签都放到这个文件夹中
path='.\\imgs'
jpgs_p=