ssd問題レコード（2）

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フレームが適用されないためだと思います。