训练的时候出现这个错误,检查标记文件的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()