好资源导航 » 文章资讯 » 对python修改xml文件的节点值方法详解

对python修改xml文件的节点值方法详解

2023-09-02 12:57:04 70

这是我的xml文件结构



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

现在想实现的是修改图像的size和目标

__author__='Sam'

importcv2
importxml.etree.ElementTreeasET
importos
importsys
importlxml
importshutil

#userinputfilespath
path="E:/test_folder"
image_path=path+"/Annotations/"#imagepathwith.jpgending
label_path=path+"/JPEGImages/"#labelpathwith.xmlending
min_size=800

defsearch_jpg_xml(image_dir,label_dir):
#findoutallofsepecifiedfile
image_ext='.jpg'
img=[fnforfninos.listdir(image_dir)iffn.endswith(image_ext)]
label_ext='.xml'
label=[fnforfninos.listdir(label_dir)iffn.endswith(label_ext)]
returnimg,label

defcopyfile():
if"Annotations_temp"inos.listdir(path):
shutil.rmtree(path+"/Annotations_temp")
if"JPEGImages_temp"inos.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)
returnsave_jpg_path,save_annotation_path

defwrite_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"notinos.listdir(path):
os.mkdir(path+"/Annotations_1")
if"JPEGImages_1"notinos.listdir(path):
os.mkdir(path+"/JPEGImages_1")
new_image_path=path+"/JPEGImages_1/"
new_annotation_path=path+"/Annotations_1/"
forindex,fileinenumerate(label):
cur_img=cv2.imread(jpg_path+img[index])
width=cur_img.shape[1]
height=cur_img.shape[0]
ifwidthheight:
new_width=int(min_size*width/height)
new_height=min_size
w_ratio=new_width/width
h_ratio=new_height/height
elifwidth==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()
fornodeinroot:
ifnode.tag=='size':
node[0].text=str(new_width)
node[1].text=str(new_height)
elifnode.tag=='object':
xmin=int(node[4][0].text)#bboxposition
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文件的节点值方法详解就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持毛票票。

声明：本文内容来源于网络，版权归原作者所有，内容由互联网用户自发贡献自行上传，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任。如果您发现有涉嫌版权的内容，欢迎发送邮件至：czq8825#qq.com（发邮件时，请将#更换为@）进行举报，并提供相关证据，一经查实，本站将立刻删除涉嫌侵权内容。

返回顶部
514930285
czq8825@qq.com

对python修改xml文件的节点值方法详解

热门推荐

随机推荐