python:批量统计xml中各类目标的数量案例_python统计xml中目标个数和裁剪的个数不一样资源-CSDN文库

70 浏览量 2021-01-20 01:41:47 上传评论收藏 116KB PDF 举报

资源推荐

资源详情

资源评论

python:批量统计批量统计xml中各类目标的数量案例中各类目标的数量案例

之前写了一个matlab的，越用越觉得麻烦，如果不同数据集要改类别数目，而且运行速度慢。所以重新写了一个Python的，

直接读取xml文件夹路径就可以，不用预先知道类别，直接能够检测出所有类别的目标名称及其对应的数量。

分享出来给大家。

代码如下：

# -*- coding:utf-8 -*-

import os

import xml.etree.ElementTree as ET

import numpy as np

np.set_printoptions(suppress=True, threshold=np.nan)

import matplotlib

from PIL import Image

def parse_obj(xml_path, filename):

tree=ET.parse(xml_path+filename)

objects=[] for obj in tree.findall('object'):

obj_struct={}

obj_struct['name']=obj.find('name').text

objects.append(obj_struct)

return objects

def read_image(image_path, filename):

im=Image.open(image_path+filename)

W=im.size[0] H=im.size[1] area=W*H

im_info=[W,H,area] return im_info

if __name__ == '__main__':

xml_path='C:/Users/nansbas/Desktop/hebin/03/'

filenamess=os.listdir(xml_path)

filenames=[] for name in filenamess:

name=name.replace('.xml','')

filenames.append(name)

recs={}

obs_shape={}

classnames=[] num_objs={}

obj_avg={}

for i,name in enumerate(filenames):

recs[name]=parse_obj(xml_path, name+ '.xml' )

for name in filenames:

for object in recs[name]:

if object['name'] not in num_objs.keys():

num_objs[object['name']]=1

else:

num_objs[object['name']]+=1

if object['name'] not in classnames:

classnames.append(object['name'])

for name in classnames:

print('{}:{}个'.format(name,num_objs[name]))

print('信息统计算完毕。')

补充知识：补充知识：Python对目标检测数据集对目标检测数据集xml文件操作（统计目标种类、数量、面积、比例等文件操作（统计目标种类、数量、面积、比例等&修改目标名字）修改目标名字）

1. 根据xml文件统计目标种类以及数量

本内容试读结束，登录后可阅读更多

下载后可阅读完整内容，剩余4页未读，立即下载

内容反馈

weixin_38621272

粉丝: 3
资源: 957

最新资源

资源上传下载、课程学习等过程中有任何疑问或建议，欢迎提出宝贵意见哦~我们会及时处理！点击此处反馈

feedback-tip