读取xml文件
假设xml文件是这样的(VOC的标注)：
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
<annotation> <folder>detection</folder> <filename>00001.png</filename> <size> <width>478</width> <height>270</height> <depth>3</depth> </size> <segmented>0</segmented> <object> <name>person</name> <pose>Unspecified</pose> <truncated>0</truncated> <difficult>0</difficult> <bndbox> <xmin>420</xmin> <ymin>125</ymin> <xmax>462</xmax> <ymax>256</ymax> </bndbox> </object> <object> <name>person</name> <pose>Unspecified</pose> <truncated>0</truncated> <difficult>0</difficult> <bndbox> <xmin>463</xmin> <ymin>125</ymin> <xmax>479</xmax> <ymax>255</ymax> </bndbox> </object> <object> <name>person</name> <pose>Unspecified</pose> <truncated>0</truncated> <difficult>0</difficult> <bndbox> <xmin>286</xmin> <ymin>97</ymin> <xmax>301</xmax> <ymax>138</ymax> </bndbox> </object> </annotation> |
读取方法如下：
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
import xml.etree.ElementTree as ET with open('a.xml', 'r') as f: tree=ET.parse(f) root = tree.getroot() root # <Element 'annotation' at 0x107577a98> list(root) # <Element 'folder' at 0x107577908> # <Element 'filename' at 0x107577ef8> # <Element 'size' at 0x107577868> # <Element 'segmented' at 0x107577d18> # <Element 'object' at 0x1075777c8> # <Element 'object' at 0x1075775e8> # <Element 'object' at 0x10741dcc8> size = root.find('size') # 获取子元素 w = int(size.find('width').text) # 读取值 h = int(size.find('height').text) for obj in root.iter('object'): # 多个元素 difficult = obj.find('difficult').text cls = obj.find('name').text |
