這是我的xml文件結構

<?xml version='1.0' encoding='utf-8'?>

JPEGImages

train_2018-05-08_1000.jpg

D:\all_data\2018-05-08\JPEGImages\train_2018-05-08_1000.jpg

Unknown

4032

3024

3

0

yl-ylhzdhmbbz-gz-hm-280g

Unspecified

0

0

1863

355

2512

902

hy-hybfbgz-hz-xcw-200ml

Unspecified

0

0

1076

1602

1648

2105

ys-zzyspyz-gz-yw-245ml

Unspecified

1

0

2017

2475

2681

3024

mn-zgl-hz-cmw-250ml

Unspecified

0

0

1849

1207

2242

2047

qc-qckf-pz-shnt-268ml

Unspecified

0

0

480

1213

1308

1544

wt-wtcyl-gz-nm-310ml

Unspecified

0

0

867

488

1527

938

現在想實現的是修改圖像的size和目標

__author__ = 'Sam'

import cv2

import xml.etree.ElementTree as ET

import os

import sys

import lxml

import shutil

#user input files path

path="E:/test_folder"

image_path = path + "/Annotations/" #image path with .jpg ending

label_path = path + "/JPEGImages/" #label path with .xml ending

min_size=800

def search_jpg_xml(image_dir,label_dir):

#find out all of sepecified file

image_ext='.jpg'

img=[fn for fn in os.listdir(image_dir) if fn.endswith(image_ext)]

label_ext='.xml'

label=[fn for fn in os.listdir(label_dir) if fn.endswith(label_ext)]

return img, label

def copyfile():

if "Annotations_temp" in os.listdir(path):

shutil.rmtree(path+"/Annotations_temp")

if "JPEGImages_temp" in os.listdir(path):

shutil.rmtree(path+"/JPEGImages_temp")

save_annotation_path=path+"/Annotations_temp/"

save_jpg_path=path+"/JPEGImages_temp/"

shutil.copytree(path + "/Annotations",save_annotation_path)

shutil.copytree(path + "/JPEGImages", save_jpg_path)

return save_jpg_path ,save_annotation_path

def write_xml_jpg(jpg_path,annotation_path):

img,label=search_jpg_xml(jpg_path,annotation_path)

sorted(img)

sorted(label)

print(img)

print(label)

if "Annotations_1" not in os.listdir(path):

os.mkdir(path+"/Annotations_1")

if "JPEGImages_1" not in os.listdir(path):

os.mkdir(path+"/JPEGImages_1")

new_image_path=path+"/JPEGImages_1/"

new_annotation_path=path+"/Annotations_1/"

for index,file in enumerate(label):

cur_img = cv2.imread(jpg_path+img[index])

width=cur_img.shape[1]

height=cur_img.shape[0]

if width

new_width=min_size

new_height=int(min_size*height/width)

w_ratio=new_width/width

h_ratio=new_height/height

elif width>height:

new_width=int(min_size*width/height)

new_height=min_size

w_ratio=new_width/width

h_ratio=new_height/height

elif width==height:

new_width=min_size

new_height=min_size

w_ratio=new_width/width

h_ratio=new_height/height

cur_img = cv2.resize(cur_img, (new_width, new_height))

cv2.imwrite(new_image_path+img[index],cur_img)

cur_xml = ET.parse(annotation_path+file)

root = cur_xml.getroot()

for node in root:

if node.tag=='size':

node[0].text=str(new_width)

node[1].text=str(new_height)

elif node.tag=='object':

xmin=int(node[4][0].text)#bbox position

ymin=int(node[4][1].text)

xmax=int(node[4][2].text)

ymax=int(node[4][3].text)

node[4][0].text=str(int(xmin*w_ratio))

node[4][1].text=str(int(ymin*h_ratio))

node[4][2].text=str(int(xmax*w_ratio))

node[4][3].text=str(int(ymax*h_ratio))

cur_xml.write(new_annotation_path+file)

shutil.rmtree(path+"/JPEGImages_temp")

shutil.rmtree(path+"/Annotations_temp")

if __name__ == "__main__":

jpg_path,annotation_path=copyfile()

write_xml_jpg(jpg_path,annotation_path)

最關鍵語句是：

node[4][3].text=str(int(ymax*h_ratio))，注意xml節點的操作是字符型！！!

以上這篇對python修改xml文件的節點值方法詳解就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持我們。

本文標題: 對python修改xml文件的節點值方法詳解

本文地址: http://www.cppcns.com/jiaoben/python/248200.html

新人創作打卡挑戰賽發博客就能抽獎！定制產品紅包拿不停！

總結

以上是生活随笔為你收集整理的python修改xml标签的值_对python修改xml文件的节点值方法详解的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。