目标检测VOC数据集根据标注文件生成txt文件

简介: 目标检测VOC数据集根据标注文件生成txt文件

Pascal VOC数据集介绍:

└── data   #数据集名字
        ├── Annotations           # 存放xml标注文件,与JPEGImages中的图片一一对应
        ├── ImageSets             # 该目录下存放的都是txt文件,txt文件中每一行包含一个图片的名称
        │   ├── Main              # 存放的是图像物体识别的数据,分为20类
        │       └── trainval.txt  # train与val的合集
        │       └── test.txt      # 用于测试的图片名称
        ├── JPEGImages            # 存放源图片

根据Annotations里面的xml标注文件,随机生成trainval.txt和test.txt,生成比例可自行调整。

# data_path为数据集路径,trainval_percent为训练验证集的比例
def get_txt(data_path,trainval_percent):
    print(data_path)
    xml_path = data_path+'/Annotations'
    print(xml_path)
    txt_path = data_path+'/ImageSets/Main'
    total_xml = os.listdir(xml_path)
    num = len(total_xml)
    list = range(num)
    tv = int(num * trainval_percent)
    trainval = random.sample(list, tv)
    ftrainval = open(os.path.join(txt_path+'/trainval.txt'), 'w')
    ftest = open(os.path.join(txt_path+'/test.txt'), 'w')
    for i in list:
        name = total_xml[i][:-4] + '\n'
        if i in trainval:
            ftrainval.write(name)
        else:
            ftest.write(name)
    ftrainval.close()
    ftest.close()


相关文章
|
XML 存储 数据格式
【30】yolov5的数据集准备 | 处理Pascal voc格式的数据集
【30】yolov5的数据集准备 | 处理Pascal voc格式的数据集
483 0
【30】yolov5的数据集准备 | 处理Pascal voc格式的数据集
|
机器学习/深度学习 监控 数据可视化
【31】yolov5的使用 | 训练Pascal voc格式的数据集
【31】yolov5的使用 | 训练Pascal voc格式的数据集
1051 0
【31】yolov5的使用 | 训练Pascal voc格式的数据集
|
XML JSON 数据中心
目标检测VOC数据集标注XML文件转EasyDL数据集标注Json格式
目标检测VOC数据集标注XML文件转EasyDL数据集标注Json格式
目标检测VOC数据集标注XML文件转EasyDL数据集标注Json格式
|
7月前
|
XML 数据格式 Python
Labelimg标注自己的数据集,及如何划分训练集和验证集,应用于Yolov5
Labelimg标注自己的数据集,及如何划分训练集和验证集,应用于Yolov5
1514 0
|
2月前
|
XML JSON 数据可视化
数据集学习笔记(六):目标检测和图像分割标注软件介绍和使用,并转换成YOLO系列可使用的数据集格式
本文介绍了labelImg和labelme两款图像标注工具的安装、使用、数据转换和验证方法,适用于目标检测和图像分割任务,支持YOLO等数据集格式。
543 2
数据集学习笔记(六):目标检测和图像分割标注软件介绍和使用,并转换成YOLO系列可使用的数据集格式
|
2月前
|
XML 计算机视觉 数据格式
数据集学习笔记(四):VOC转COCO数据集并据txt中图片的名字批量提取对应的图片并保存到另一个文件夹
这篇文章介绍了如何将VOC数据集转换为COCO数据集的格式,并通过Python脚本根据txt文件中列出的图片名称批量提取对应的图片并保存到另一个文件夹。
32 3
|
4月前
分割数据集,并对数据集进行预处理
【8月更文挑战第8天】分割数据集,并对数据集进行预处理。
39 1
|
7月前
|
机器学习/深度学习 PyTorch 算法框架/工具
数据集 VOC转YOLO格式2
数据集 VOC转YOLO格式
|
7月前
安装并使用labelImg标注数据集,yolo,VOC格式
安装并使用labelImg标注数据集,yolo,VOC格式
223 0
|
XML 数据可视化 数据格式
【数据集显示标注】VOC文件结构+数据集标注可视化+代码实现
【数据集显示标注】VOC文件结构+数据集标注可视化+代码实现
456 0