##### 根据同文件夹一个xml,生成其他图像的xml #####
import os
import xml.etree.ElementTree as ET
from xml.etree.ElementTree import parse, Element
import shutil
pcb_list=os.listdir('./')
print(pcb_list)
for folder in pcb_list:
if folder.endswith('py'):
pass
else:
new_folder='./'+folder+'/'
new_folder_list=os.listdir(new_folder)
for im in new_folder_list:
# print(im)
if im.endswith('.xml'):
# print(im)
file_xml = im
for im in new_folder_list:
if im.split('.')[0] == file_xml.split('.')[0]:
pass
else:
new_xml='%s.xml'%(im.split('.')[0])
shutil.copyfile(new_folder+file_xml,new_folder+new_xml)
dom = parse(new_folder+new_xml)
root = dom.getroot()
for obj in root.iter('annotation'):
obj.find('filename').text = im
print(obj.find('filename').text)
# print(obj.find('path').text)
dom.write(new_folder+new_xml, xml_declaration=True)
python-根据同文件夹一个xml,生成其他图像的xml(目标检测)
最新推荐文章于 2022-10-20 16:34:39 发布