xml缺少size标签解决方法

import os
import xml.dom.minidom
from xml.etree.ElementTree import ElementTree, Element
import cv2









path = r"C:\Users\gff20\Desktop\V1"

def getFileName1(path,suffix,suffix2):
    # 获取指定目录下的所有指定后缀的文件名
    input_template_All=[]
    f_list = os.listdir(path)#返回文件名
    for i in f_list:
        if os.path.splitext(i)[1] == suffix:
            input_template_All.append(i)
            print(i)

            img = cv2.imread(path + "/" + i)

            high, width = img.shape[:2]
            print(high, width)

        if os.path.splitext(i)[1] == suffix2:
            input_template_All.append(i)

            tree = ElementTree()

            tree.parse(path + '/' + i)

            root = tree.getroot()

            element = Element('size')

            one = Element('width')
            one.text = str(width)
            element.append(one)

            two = Element('height')
            two.text = str(high)
            element.append(two)

            three = Element('depth')
            three.text = '3'
            element.append(three)

            root.append(element)
            tree.write(path + '/' + i,encoding='utf-8', xml_declaration=True)

    return input_template_All

input_template_All1 = getFileName1(path, '.jpg','.xml')