ssd问题记录（二）

1、自己制作pasval voc格式数据集，比赛官方给的xml文件中没有size属性，我根据以下代码自己添加进去

from lxml import etree
from tqdm import tqdm
import os
def add_pic_attr(root):
  	object = etree.Element("size")
    widthn =  etree.SubElement(object,"width")
    widthn.text = str(1920)
    object.append(widthn)
    heightn =  etree.SubElement(object,"height")
    heightn.text = str(1080)
    object.append(heightn)
    depthn =  etree.SubElement(object,"depth")
    depthn.text = str(3)
    object.append(depthn)
    root.getroot().append(object)

if __name__ == '__main__':
    path = r'E:\TEMP'
    xml_files = [os.path.join(rootdir, file) for rootdir, _, files in os.walk(path) for file in files if
                 (file.endswith('.xml'))]
    for xml in tqdm(xml_files):
       # edit_xml(xml)  # 修改xml文件

        parser = etree.XMLParser(remove_blank_text=True)#
        root=etree.parse(xml,parser)
        add_pic_attr(root)
        tree = etree.ElementTree(root.getroot())
        tree.write(xml,pretty_print=True, xml_declaration=False, encoding='utf-8')

2、
运行时遇到这种问题，网上有人说是classes参数不匹配，我尝试修改几次后都不行。我猜测是因为训练输入图片尺寸是1920*1080与ssd框架不适用。