DDR爱好者之家 Design By 杰米
这是我的xml文件结构
<"htmlcode">__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<height: 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文件的节点值方法详解就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。
DDR爱好者之家 Design By 杰米
广告合作:本站广告合作请联系QQ:858582 申请时备注:广告合作(否则不回)
免责声明:本站资源来自互联网收集,仅供用于学习和交流,请遵循相关法律法规,本站一切资源不代表本站立场,如有侵权、后门、不妥请联系本站删除!
免责声明:本站资源来自互联网收集,仅供用于学习和交流,请遵循相关法律法规,本站一切资源不代表本站立场,如有侵权、后门、不妥请联系本站删除!
DDR爱好者之家 Design By 杰米
暂无评论...
更新日志
2024年11月27日
2024年11月27日
- 凤飞飞《我们的主题曲》飞跃制作[正版原抓WAV+CUE]
- 刘嘉亮《亮情歌2》[WAV+CUE][1G]
- 红馆40·谭咏麟《歌者恋歌浓情30年演唱会》3CD[低速原抓WAV+CUE][1.8G]
- 刘纬武《睡眠宝宝竖琴童谣 吉卜力工作室 白噪音安抚》[320K/MP3][193.25MB]
- 【轻音乐】曼托凡尼乐团《精选辑》2CD.1998[FLAC+CUE整轨]
- 邝美云《心中有爱》1989年香港DMIJP版1MTO东芝首版[WAV+CUE]
- 群星《情叹-发烧女声DSD》天籁女声发烧碟[WAV+CUE]
- 刘纬武《睡眠宝宝竖琴童谣 吉卜力工作室 白噪音安抚》[FLAC/分轨][748.03MB]
- 理想混蛋《Origin Sessions》[320K/MP3][37.47MB]
- 公馆青少年《我其实一点都不酷》[320K/MP3][78.78MB]
- 群星《情叹-发烧男声DSD》最值得珍藏的完美男声[WAV+CUE]
- 群星《国韵飘香·贵妃醉酒HQCD黑胶王》2CD[WAV]
- 卫兰《DAUGHTER》【低速原抓WAV+CUE】
- 公馆青少年《我其实一点都不酷》[FLAC/分轨][398.22MB]
- ZWEI《迟暮的花 (Explicit)》[320K/MP3][57.16MB]