.xml转.txt文件并划分train/val数据集

访风景于崇阿

已于 2023-02-08 21:42:40 修改

阅读量168

点赞数

分类专栏：深度学习学习笔记文章标签： xml

于 2023-02-08 21:40:28 首次发布

本文链接：https://blog.csdn.net/starlight1231/article/details/128943944

版权

该博客介绍了一种将XML数据集转换为TXT文件并按比例划分为训练集和验证集的方法，具体应用在人脸口罩检测数据集Face Mask Detection上。处理后的数据集结构清晰，包含单独的训练和验证数据目录。

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

一种简单的数据集处理方法，实现的功能是：

xml文件转txt文件
按比例划分为训练集和验证集

用的数据集是人脸口罩检测数据集: Face Mask Detection

处理效果

原始数据集存放：

face_devkit
└─ face-mask-detection
├─ annotations
├─ images

调整后：
face_devkit
├─ face-mask-detection
│ ├─ annotations
│ ├─ images
│
├─ iamges
│ ├─ train
│ ├─ val
│
└─ labels
├─ train # 下面放训练集标签
├─ val # 下面放验证集标签

代码

代码简单易读，所以记录一下：

"""
Face Mask Detection 数据集_2023.2.8

"""
import xml.etree.ElementTree as ET
import pickle
import os
from os import listdir, getcwd
from os.path import join
import random
from shutil import copyfile

# classes = ["hat", "person"]
classes = ["with_mask", "without_mask", "mask_weared_incorrect"]

TRAIN_RATIO = 80  # 训练集验证集划分比例


def clear_hidden_files(path):
    dir_list = os.listdir(path)
    for i in dir_list:
        abspath = os.path.join(os.path.abspath(path), i)
        if os.path.isfile(abspath):
            if i.startswith("._"):
                os.remove(abspath)