python:批量統(tǒng)計xml中各類目標的數(shù)量案例
更新時間：2020年03月10日 08:45:29 作者：南石北岸生
這篇文章主要介紹了python:批量統(tǒng)計xml中各類目標的數(shù)量案例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
之前寫了一個matlab的，越用越覺得麻煩，如果不同數(shù)據(jù)集要改類別數(shù)目，而且運行速度慢。所以重新寫了一個Python的，直接讀取xml文件夾路徑就可以，不用預先知道類別，直接能夠檢測出所有類別的目標名稱及其對應的數(shù)量。
分享出來給大家。
代碼如下：
# -*- 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('信息統(tǒng)計算完畢。')
補充知識：Python對目標檢測數(shù)據(jù)集xml文件操作（統(tǒng)計目標種類、數(shù)量、面積、比例等&修改目標名字）
1. 根據(jù)xml文件統(tǒng)計目標種類以及數(shù)量
# -*- coding:utf-8 -*-
#根據(jù)xml文件統(tǒng)計目標種類以及數(shù)量
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='/home/dlut/網(wǎng)絡/make_database/數(shù)據(jù)集——合集/VOCdevkit/VOC2018/Annotations/'
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('信息統(tǒng)計算完畢。')
2.根據(jù)xml文件統(tǒng)計目標的平均長度、寬度、面積以及每一個目標在原圖中的占比
# -*- coding:utf-8 -*-
#統(tǒng)計
# 計算每一個目標在原圖中的占比
# 計算目標的平均長度、
# 計算平均寬度，
# 計算平均面積、
# 計算目標平均占比
import os
import xml.etree.ElementTree as ET
import numpy as np
#np.set_printoptions(suppress=True, threshold=np.nan) #10,000,000
np.set_printoptions(suppress=True, threshold=10000000) #10,000,000
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
bbox = obj.find('bndbox')
obj_struct['bbox'] = [int(bbox.find('xmin').text),
int(bbox.find('ymin').text),
int(bbox.find('xmax').text),
int(bbox.find('ymax').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__':
image_path = '/home/dlut/網(wǎng)絡/make_database/數(shù)據(jù)集——合集/VOCdevkit/VOC2018/JPEGImages/'
xml_path = '/home/dlut/網(wǎng)絡/make_database/數(shù)據(jù)集——合集/VOCdevkit/VOC2018/Annotations/'
filenamess = os.listdir(xml_path)
filenames = []
for name in filenamess:
name = name.replace('.xml', '')
filenames.append(name)
print(filenames)
recs = {}
ims_info = {}
obs_shape = {}
classnames = []
num_objs={}
obj_avg = {}
for i, name in enumerate(filenames):
print('正在處理 {}.xml '.format(name))
recs[name] = parse_obj(xml_path, name + '.xml')
print('正在處理 {}.jpg '.format(name))
ims_info[name] = read_image(image_path, name + '.jpg')
print('所有信息收集完畢。')
print('正在處理信息......')
for name in filenames:
im_w = ims_info[name][0]
im_h = ims_info[name][1]
im_area = ims_info[name][2]
for object in recs[name]:
if object['name'] not in num_objs.keys():
num_objs[object['name']] = 1
else:
num_objs[object['name']] += 1
#num_objs += 1
ob_w = object['bbox'][2] - object['bbox'][0]
ob_h = object['bbox'][3] - object['bbox'][1]
ob_area = ob_w * ob_h
w_rate = ob_w / im_w
h_rate = ob_h / im_h
area_rate = ob_area / im_area
if not object['name'] in obs_shape.keys():
obs_shape[object['name']] = ([[ob_w,
ob_h,
ob_area,
w_rate,
h_rate,
area_rate]])
else:
obs_shape[object['name']].append([ob_w,
ob_h,
ob_area,
w_rate,
h_rate,
area_rate])
if object['name'] not in classnames:
classnames.append(object['name']) # 求平均
for name in classnames:
obj_avg[name] = (np.array(obs_shape[name]).sum(axis=0)) / num_objs[name]
print('{}的情況如下：*******\n'.format(name))
print(' 目標平均W={}'.format(obj_avg[name][0]))
print(' 目標平均H={}'.format(obj_avg[name][1]))
print(' 目標平均area={}'.format(obj_avg[name][2]))
print(' 目標平均與原圖的W比例={}'.format(obj_avg[name][3]))
print(' 目標平均與原圖的H比例={}'.format(obj_avg[name][4]))
print(' 目標平均原圖面積占比={}\n'.format(obj_avg[name][5]))
print('信息統(tǒng)計計算完畢。')
3.修改xml文件中某個目標的名字為另一個名字
#修改xml文件中的目標的名字，
import os, sys
import glob
from xml.etree import ElementTree as ET
# 批量讀取Annotations下的xml文件
# per=ET.parse(r'C:\Users\rockhuang\Desktop\Annotations\000003.xml')
xml_dir = r'/home/dlut/網(wǎng)絡/make_database/數(shù)據(jù)集——合集/VOCdevkit/VOC2018/Annotations'
xml_list = glob.glob(xml_dir + '/*.xml')
for xml in xml_list:
print(xml)
per = ET.parse(xml)
p = per.findall('/object')
for oneper in p: # 找出person節(jié)點
child = oneper.getchildren()[0] # 找出person節(jié)點的子節(jié)點
if child.text == 'PinNormal': #需要修改的名字
child.text = 'normal bolt' #修改成什么名字
if child.text == 'PinDefect': #需要修改的名字
child.text = 'defect bolt-1' #修改成什么名字
per.write(xml)
print(child.tag, ':', child.text)
以上這篇python:批量統(tǒng)計xml中各類目標的數(shù)量案例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
python中強制關(guān)閉線程與協(xié)程與進程方法
python使用中多線程、多進程、多協(xié)程使用是比較常見的。那么如果在多線程等的使用，我們這個時候我們想從外部強制殺掉該線程請問如何操作？這篇文章帶你介紹，感興趣的同學可以參考閱讀2023-03-03
python實現(xiàn)遞歸查找某個路徑下所有文件中的中文字符
這篇文章主要為大家詳細介紹了python實現(xiàn)遞歸查找某個路徑下所有文件中的中文字符，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-08-08
這篇文章主要介紹了wxPython+Matplotlib繪制折線圖表，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-11-11
本文給大家匯總介紹了下幾種使用Python編寫簡單爬蟲的方法和代碼，非常的不錯，這里分享給大家，希望大家能夠喜歡。2016-03-03
Python的線程使用隊列Queue來改造轉(zhuǎn)賬場景
前篇我們了隊列Queue和轉(zhuǎn)賬場景這次趁熱學委展示一下使用隊列解決轉(zhuǎn)賬場景的問題，這篇文章主要介紹了Python的線程使用隊列來改造轉(zhuǎn)賬場景,需要的朋友可以參考一下2022-02-02
本文主要介紹了使用Python第三方庫發(fā)送電子郵件的示例代碼，文中通過示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2021-12-12
這篇文章主要介紹了pyinstaller打包后,配置文件無法正常讀取的解決方案，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-02-02
這篇文章主要介紹了Python最長公共子串算法,實例分析了Python字符串操作的技巧,需要的朋友可以參考下2015-03-03
python框架Django實戰(zhàn)商城項目之工程搭建過程圖文詳解
這篇文章主要介紹了python框架Django實戰(zhàn)商城項目之工程搭建過程,這個項目很像京東商城，項目開發(fā)采用前后端不分離的模式，本文通過圖文并茂的形式給大家介紹的非常詳細，需要的朋友可以參考下2020-03-03
這篇文章主要為大家詳細介紹了pygame庫實現(xiàn)俄羅斯方塊小游戲，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-10-10