'NoneType' object has no attribute 'shape' 批量修改xml

训练的时候出现这个错误，检查标记文件的xml的文件名后缀不对，所以批量修改xml。把png改成jpg

#!/usr/bin/python
# -*- coding: UTF-8 -*-
import os
import os.path
from xml.etree.ElementTree import parse, Element


def test():
    path = "C:/Users/Administrator/Downloads/frcnn1/Annotations"
    files = os.listdir(path)  # 得到文件夹下所有文件名称
    s = []
    for xmlFile in files:  # 遍历文件夹
        newStr = os.path.join(path, xmlFile)
        print(newStr)
        dom = parse(newStr)  ###最核心的部分,路径拼接,输入的是具体路径
        root = dom.getroot()


        image_name=root.find('filename').text
        new_image_name=image_name[:-3]+'jpg'
        print(new_image_name)
        root.find('filename').text = new_image_name

        dom.write(newStr, xml_declaration=True)
        pass


if __name__ == '__main__':
    test()