1.python解析xml

img_path='./data/001.tif'xml_path='./xml/001.xml'img=cv2.imread(img_path)# cv2.imshow('img', img)# cv2.waitKey(0)print(img.shape)try:xmlp = ET.XMLParser(encoding="utf-8")tree = ET.parse(xml_path, parser=xmlp)root = tree.getroot()print(tree)print(root)except Exception as e:print(e, xml_path)content_info={}for i in tree.iterfind('.//正文組/正文'):print("i.attrib['篇目號']", i.attrib['篇目號'])for j in i.findall('篇目坐標'):print(j.text)print(type(j.text))content_info[i.attrib['篇目號']] =j.textprint('content_info=',content_info)

若要更改xml內(nèi)容，可以

new_point='1,2,3,4' j.text = new_point tree.write('output_test.xml', encoding='utf-8')

2.?得到pascal voc xml格式用于目標檢測

img_dir = "./images/train" csv_dir = "./train_xml.csv" xml_dir = "./Annotations" if not os.path.exists(xml_dir):os.mkdir(xml_dir)imgs_path_Lists = [os.path.join(img_dir,i) for i in os.listdir(img_dir)]df=pd.read_csv(csv_dir).copy() df_value=df.values print(df_value.shape) coord_h,coord_w=df_value.shape print(df_value[:2]) for img_path_List in imgs_path_Lists:im = Image.open(img_path_List)width, height = im.sizeimg_name=img_path_List.split('/')[-1]# write in xml filexml_file = open((xml_dir + '/' + img_name.split('.jpg')[0] + '.xml'), 'w')xml_file.write('<annotation>\n')xml_file.write(' <folder>steel</folder>\n')xml_file.write(' <filename>' + img_name + '</filename>\n')xml_file.write(' <size>\n')xml_file.write(' <width>' + str(width) + '</width>\n')xml_file.write(' <height>' + str(height) + '</height>\n')xml_file.write(' <depth>3</depth>\n')xml_file.write(' </size>\n')for i in df_value:if i[0].split('/')[-1]==img_name:xmin=i[1]ymin=i[2]xmax=i[3]ymax=i[4]class_name=i[-1]# write the region of image on xml filexml_file.write(' <object>\n')xml_file.write(' <name>' + class_name + '</name>\n')xml_file.write(' <pose>Unspecified</pose>\n')xml_file.write(' <truncated>0</truncated>\n')xml_file.write(' <difficult>0</difficult>\n')xml_file.write(' <bndbox>\n')xml_file.write(' <xmin>' + str(xmin) + '</xmin>\n')xml_file.write(' <ymin>' + str(ymin) + '</ymin>\n')xml_file.write(' <xmax>' + str(xmax) + '</xmax>\n')xml_file.write(' <ymax>' + str(ymax) + '</ymax>\n')xml_file.write(' </bndbox>\n')xml_file.write(' </object>\n')xml_file.write('</annotation>')

3.美化xml

原始xml如下

<?xml version='1.0' encoding='utf-8'?> <lab><person name='Blue'><age>22</age><gender>male</gender></person><person name='Yellow'><age>22</age><gender>female</gender></person></lab>

代碼:

def prettyXml(element, indent, newline, level = 0): # elemnt為傳進來的Elment類，參數(shù)indent用于縮進，newline用于換行if element: # 判斷element是否有子元素if element.text == None or element.text.isspace(): # 如果element的text沒有內(nèi)容element.text = newline + indent * (level + 1)else:element.text = newline + indent * (level + 1) + element.text.strip() + newline + indent * (level + 1)#else: # 此處兩行如果把注釋去掉，Element的text也會另起一行#element.text = newline + indent * (level + 1) + element.text.strip() + newline + indent * leveltemp = list(element) # 將elemnt轉成listfor subelement in temp:if temp.index(subelement) < (len(temp) - 1): # 如果不是list的最后一個元素，說明下一個行是同級別元素的起始，縮進應一致subelement.tail = newline + indent * (level + 1)else: # 如果是list的最后一個元素， 說明下一行是母元素的結束，縮進應該少一個subelement.tail = newline + indent * levelprettyXml(subelement, indent, newline, level = level + 1) # 對子元素進行遞歸操作 def testXML():from xml.etree import ElementTree as ETtree = ET.parse('test.xml')root = tree.getroot()print(root)# #查看tag和attrib# for person in root:# print(person.tag, person.attrib)ZF_COORD = ET.Element('字符坐標')ZWZ = ET.SubElement(ZF_COORD, '正文組')#這里需要把001變成需要的篇目號ZW = ET.SubElement(ZWZ , '正文',{'篇目號':"001"})#欄目LM=ET.SubElement(ZW, '欄目')LM.text = '12,12,334,123;12,34,45,56'#引題YT=ET.SubElement(ZW,'引題')YT.text='12,12,334,123;12,34,45,56'# 標題BT = ET.SubElement(ZW, '標題')BT.text = '12,12,334,123;12,34,45,56'# 副題FT = ET.SubElement(ZW, '副題')FT.text = '12,12,334,123;12,34,45,56'# 小標題XBT = ET.SubElement(ZW, '小標題')XBT.text = '12,12,334,123;12,34,45,56'# 作者author = ET.SubElement(ZW, '作者')author.text = '12,12,334,123;12,34,45,56'# # 向根節(jié)點添加新的子節(jié)點?root.append(ZF_COORD)# 寫入?tree.write('./sample.xml', encoding="utf-8",xml_declaration=True)#美化作用tree = ET.parse('./sample.xml') # 解析test.xml這個文件，該文件內(nèi)容如上文root = tree.getroot() # 得到根元素，Element類prettyXml(root, '\t', '\n') # 執(zhí)行美化方法# ET.dump(root)tree.write('./sample.xml', encoding="utf-8",xml_declaration=True)

新生成xml如下

<?xml version='1.0' encoding='utf-8'?> <lab><person name="Blue"><age>22</age><gender>male</gender></person><person name="Yellow"><age>22</age><gender>female</gender></person><字符坐標><正文組><正文 篇目號="001"><欄目>12,12,334,123;12,34,45,56</欄目><引題>12,12,334,123;12,34,45,56</引題><標題>12,12,334,123;12,34,45,56</標題><副題>12,12,334,123;12,34,45,56</副題><小標題>12,12,334,123;12,34,45,56</小標題><作者>12,12,334,123;12,34,45,56</作者></正文></正文組></字符坐標> </lab>

?

總結

以上是生活随笔為你收集整理的python解析xml+得到pascal voc xml格式用于目标检测+美化xml的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。