import os
import xml.etree.ElementTree as ET
# 设置需要调整框大小xml文件夹路径
def edit_xml(xml_file,xminimum,yminimum,xmaximum,ymaximum):
all_xml_file = os.path.join(path, xml_file)
tree = ET.parse(all_xml_file)
print(all_xml_file)
objs = tree.findall('object')
for ix, obj in enumerate(objs):
obj_bnd = obj.find('bndbox')
obj_xmin = obj_bnd.find('xmin')
obj_ymin = obj_bnd.find('ymin')
obj_xmax = obj_bnd.find('xmax')
obj_ymax = obj_bnd.find('ymax')
xmin = float(obj_xmin.text)
ymin = float(obj_ymin.text)
xmax = float(obj_xmax.text)
ymax = float(obj_ymax.text)
obj_xmin.text = str(round(xmin-xminimum)) #-左
obj_ymin.text = str(round(ymin-yminimum)) #-上
obj_xmax.text = str(round(xmax+xmaximum)) #+右
obj_ymax.text = str(round(ymax+ymaximum)) #+下
tree.write(all_xml_file, method='xml', encoding='utf-8') # 覆盖更新xml文件
def Adjustborder(path,a,b,c,d):
files = os.listdir(path) # 获取文件名列表
for i, file in enumerate(files):
if file.endswith('.xml'):
edit_xml(file,a,b,c,d)
#xml文件路径
path = r'F:\xml'
#调整像素值,正数就是扩大,负值就是减少
xminimum=3
yminimum=3
xmaximum=3
ymaximum=3
Adjustborder(path,xminimum,yminimum,xmaximum,ymaximum)