【目标检测】小脚本：根据xml文件统计类别数

zstar

发布于 2022-09-22 16:18:29

51400

代码可运行

文章被收录于专栏：往期博文往期博文

运行总次数：0

代码可运行

问题场景

搜到一个目标检测数据集，但是别人没有详细说明到底有多少类别以及各类别名称，这时候就需要查询xml的标注文件来获取这两个信息。

脚本代码

import xml.etree.ElementTree as ET
import numpy as np
import os

if __name__ == '__main__':
    xmlfilepath = 'D:/Dataset/mydata/Annotations'  # xml文件存放地址
    classes = []
    for i in os.listdir(xmlfilepath):
        in_file = open(xmlfilepath + '/' + i)
        tree = ET.parse(in_file)
        root = tree.getroot()
        for obj in root.iter('object'):
            cls = obj.find('name').text
            classes.append(cls)
    print("总类别数:{}".format(len(np.unique(classes))))
    print(np.unique(classes))

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2022-08-16，如有侵权请联系 cloudcommunity@tencent.com 删除

xml