欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

python:批量統(tǒng)計xml中各類目標(biāo)的數(shù)量案例

 更新時間:2020年03月10日 08:45:29   作者:南石北岸生  
這篇文章主要介紹了python:批量統(tǒng)計xml中各類目標(biāo)的數(shù)量案例,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧

之前寫了一個matlab的,越用越覺得麻煩,如果不同數(shù)據(jù)集要改類別數(shù)目,而且運行速度慢。所以重新寫了一個Python的,直接讀取xml文件夾路徑就可以,不用預(yù)先知道類別,直接能夠檢測出所有類別的目標(biāo)名稱及其對應(yīng)的數(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對目標(biāo)檢測數(shù)據(jù)集xml文件操作(統(tǒng)計目標(biāo)種類、數(shù)量、面積、比例等&修改目標(biāo)名字)

1. 根據(jù)xml文件統(tǒng)計目標(biāo)種類以及數(shù)量

# -*- coding:utf-8 -*-
#根據(jù)xml文件統(tǒng)計目標(biāo)種類以及數(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)絡(luò)/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)計目標(biāo)的平均長度、寬度、面積以及每一個目標(biāo)在原圖中的占比

# -*- coding:utf-8 -*-
#統(tǒng)計
# 計算每一個目標(biāo)在原圖中的占比
# 計算目標(biāo)的平均長度、
# 計算平均寬度,
# 計算平均面積、
# 計算目標(biāo)平均占比

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)絡(luò)/make_database/數(shù)據(jù)集——合集/VOCdevkit/VOC2018/JPEGImages/'
 xml_path = '/home/dlut/網(wǎng)絡(luò)/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(' 目標(biāo)平均W={}'.format(obj_avg[name][0]))
  print(' 目標(biāo)平均H={}'.format(obj_avg[name][1]))
  print(' 目標(biāo)平均area={}'.format(obj_avg[name][2]))
  print(' 目標(biāo)平均與原圖的W比例={}'.format(obj_avg[name][3]))
  print(' 目標(biāo)平均與原圖的H比例={}'.format(obj_avg[name][4]))
  print(' 目標(biāo)平均原圖面積占比={}\n'.format(obj_avg[name][5]))
 print('信息統(tǒng)計計算完畢。')

3.修改xml文件中某個目標(biāo)的名字為另一個名字

#修改xml文件中的目標(biāo)的名字,
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)絡(luò)/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中各類目標(biāo)的數(shù)量案例就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。

相關(guān)文章

  • python中強制關(guān)閉線程與協(xié)程與進(jìn)程方法

    python中強制關(guān)閉線程與協(xié)程與進(jìn)程方法

    python使用中多線程、多進(jìn)程、多協(xié)程使用是比較常見的。那么如果在多線程等的使用,我們這個時候我們想從外部強制殺掉該線程請問如何操作?這篇文章帶你介紹,感興趣的同學(xué)可以參考閱讀
    2023-03-03
  • python實現(xiàn)遞歸查找某個路徑下所有文件中的中文字符

    python實現(xiàn)遞歸查找某個路徑下所有文件中的中文字符

    這篇文章主要為大家詳細(xì)介紹了python實現(xiàn)遞歸查找某個路徑下所有文件中的中文字符,具有一定的參考價值,感興趣的小伙伴們可以參考一下
    2019-08-08
  • wxPython+Matplotlib繪制折線圖表

    wxPython+Matplotlib繪制折線圖表

    這篇文章主要介紹了wxPython+Matplotlib繪制折線圖表,文中示例代碼介紹的非常詳細(xì),具有一定的參考價值,感興趣的小伙伴們可以參考一下
    2019-11-11
  • python編寫簡單爬蟲資料匯總

    python編寫簡單爬蟲資料匯總

    本文給大家匯總介紹了下幾種使用Python編寫簡單爬蟲的方法和代碼,非常的不錯,這里分享給大家,希望大家能夠喜歡。
    2016-03-03
  • Python的線程使用隊列Queue來改造轉(zhuǎn)賬場景

    Python的線程使用隊列Queue來改造轉(zhuǎn)賬場景

    前篇我們了隊列Queue和轉(zhuǎn)賬場景這次趁熱學(xué)委展示一下使用隊列解決轉(zhuǎn)賬場景的問題,這篇文章主要介紹了Python的線程使用隊列來改造轉(zhuǎn)賬場景,需要的朋友可以參考一下
    2022-02-02
  • 使用Python第三方庫發(fā)送電子郵件的示例代碼

    使用Python第三方庫發(fā)送電子郵件的示例代碼

    本文主要介紹了使用Python第三方庫發(fā)送電子郵件的示例代碼,文中通過示例代碼介紹的非常詳細(xì),具有一定的參考價值,感興趣的小伙伴們可以參考一下
    2021-12-12
  • pyinstaller打包后,配置文件無法正常讀取的解決

    pyinstaller打包后,配置文件無法正常讀取的解決

    這篇文章主要介紹了pyinstaller打包后,配置文件無法正常讀取的解決方案,具有很好的參考價值,希望對大家有所幫助。如有錯誤或未考慮完全的地方,望不吝賜教
    2022-02-02
  • Python最長公共子串算法實例

    Python最長公共子串算法實例

    這篇文章主要介紹了Python最長公共子串算法,實例分析了Python字符串操作的技巧,需要的朋友可以參考下
    2015-03-03
  • python框架Django實戰(zhàn)商城項目之工程搭建過程圖文詳解

    python框架Django實戰(zhàn)商城項目之工程搭建過程圖文詳解

    這篇文章主要介紹了python框架Django實戰(zhàn)商城項目之工程搭建過程,這個項目很像京東商城,項目開發(fā)采用前后端不分離的模式,本文通過圖文并茂的形式給大家介紹的非常詳細(xì),需要的朋友可以參考下
    2020-03-03
  • pygame庫實現(xiàn)俄羅斯方塊小游戲

    pygame庫實現(xiàn)俄羅斯方塊小游戲

    這篇文章主要為大家詳細(xì)介紹了pygame庫實現(xiàn)俄羅斯方塊小游戲,文中示例代碼介紹的非常詳細(xì),具有一定的參考價值,感興趣的小伙伴們可以參考一下
    2019-10-10

最新評論