python提取voc数据集person类头部框（坐标）

最新推荐文章于 2021-11-28 21:09:49 发布

zcy904983

最新推荐文章于 2021-11-28 21:09:49 发布

阅读量513

点赞数

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

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

版权

这篇博客介绍了如何使用Python的xml.dom.minidom库从VOC数据集的XML文件中提取person类别的头部框坐标，并将这些信息保存到TXT文件中。

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

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

import xml.dom.minidom
import os

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

    dom1 = xml.dom.minidom.parse(ann_filepath + '\\' + file)        #打开xml文件
    root = dom1.documentElement              # 得到文档元素对象
    print(root.nodeName)
    parts=root.getElementsByTagName("part")
    for part in par