1.Créez moi-même le jeu de données au format pasval voc. Il n'y a pas d'attribut de taille dans le fichier xml fourni par le responsable du concours. Je l'ajoute moi-même selon le code suivant
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. En rencontrant
ce genre de problème, certaines personnes sur Internet disent que le paramètre classes ne correspond pas. Après avoir essayé de le modifier plusieurs fois, cela n'a pas fonctionné. Je suppose que c'est parce que la taille de l'image d'entrée d'entraînement est de 1920 * 1080 et que le cadre ssd n'est pas applicable.