提取VOC数据person类的全身标签（坐标）

最新推荐文章于 2023-06-06 19:20:10 发布

zcy904983

最新推荐文章于 2023-06-06 19:20:10 发布

阅读量450

点赞数

分类专栏：数据处理文章标签： xml python

本文链接：https://blog.csdn.net/zcy904983/article/details/105775911

版权

本文介绍如何从VOC数据集的XML文件中提取person类别的全身标签坐标，并将其保存到TXT文件中。使用了两种不同的Python解析库，即xml.dom.minidom和xml.etree.ElementTree。

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

import xml.dom.minidom # 第一种解析
import os

打开xml文件，存到TXT文件里面(处理person类的全身标签)

import xml.etree.ElementTree as ET # 第二种解析
if name == ‘main’:
ann_filepath = ‘D:/University/Data/VOCdevkit/VOC2012/Annotations_ssd/’
ann_savepath = ‘D:/University/Data/VOCdevkit/VOC2012/Annotations_result/’
if not os.path.exists(ann_savepath): # 创建文件夹
os.mkdir(ann_savepath)

for file in os.listdir(ann_filepath):
    fp = open(ann_filepath + '\\' + file)       # 读取原文件
    newfile = file.replace('.xml','.txt')
    ann_savefile = ann_savepath + newfile       # 保存文件
    fp_w = open(ann_savefile, 'w')

# tree = ET.parse('D:/University/Data/VOCdevkit/VOC2012/Annotations_ssd/2007_000272.xml')
tree = ET.parse(ann_filepath + '\\' + file)
root1 = tree.getroot()

for obj in r