欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

ssd问题记录(二)

程序员文章站 2024-03-17 11:42:40
...

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、ssd问题记录(二)
运行时遇到这种问题,网上有人说是classes参数不匹配,我尝试修改几次后都不行。我猜测是因为训练输入图片尺寸是1920*1080与ssd框架不适用。

相关标签: ssd