【目标检测】python代码获取xml文件中所有标签名称

1.需求背景:

现在手上有一份目标检测数据集,这个数据集很大,总共10万张图片,对应10万个xml文件,我们可以使用标注工具labelimg打开查看数据集,人工统计数据集中标签的名称,但是,数据集太大,人工统计费时费力,而且很容易出出错,因此需要写一个脚本代码实现获取xml文件中所有标签名称。

2.需求分析

查看xml文件中的内容:

【目标检测】python代码获取xml文件中所有标签名称_第1张图片
我们主要提取xml文件中的类别名称


    ore carrier 

总结:通过遍历xml文件中的name,将name中的类别添加到一个列表,再将列表去重,得到整个数据集的类别名称。

3.代码实现

# -*- coding: utf-8 -*-
"""
Created on Mon Nov 28 08:26:28 2022

@author: zqq
"""


import xml.etr

你可能感兴趣的:(深度学习--目标检测,python,目标检测,xml)