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、
运行时遇到这种问题,网上有人说是classes参数不匹配,我尝试修改几次后都不行。我猜测是因为训练输入图片尺寸是1920*1080与ssd框架不适用。
上一篇: IO之——递归方法删除文件夹和文件
下一篇: LeetCode-复原IP地址