python - darkflow - 属性错误 : 'NoneType' object has no attribute 'find'

Question

darkflow 和 python 的新手。我正在尝试使用 https://github.com/thtrieu/darkflow 训练我自己的数据集和 yolo 权重。我目前有错误:

AttributeError: 'NoneType' object has no attribute 'find'

执行pascal_voc_clean_xml.py时第41行:'w = (imsize.find('width').text)'

这是来自 https://github.com/thtrieu/darkflow/blob/master/darkflow/utils/pascal_voc_clean_xml.py 的部分代码:

in_file = open(file)
    tree=ET.parse(in_file)
    root = tree.getroot()
    jpg = str(root.find('filename').text)
    imsize = root.find('size')
    w = int(imsize.find('width').text)
    h = int(imsize.find('height').text)
    all = list()

    for obj in root.iter('object'):
            current = list()
            name = obj.find('name').text
            if name not in pick:
                    continue

            xmlbox = obj.find('bndbox')
            xn = int(float(xmlbox.find('xmin').text))
            xx = int(float(xmlbox.find('xmax').text))
            yn = int(float(xmlbox.find('ymin').text))
            yx = int(float(xmlbox.find('ymax').text))
            current = [name,xn,yn,xx,yx]
            all += [current]

    add = [[jpg, [w, h, all]]]
    dumps += add
    in_file.close()

这是我的 xml 文件:

我知道“find”是 python 中 ElementTree.py 的一个函数，但为什么“find”函数不能正常工作？

Answer 1

检查注释中的数据(图像文件夹)

for i, file in enumerate(annotations):

也许 image00001.jpg 的 xml 文件与您的错误代码无关。

其他一些 xml 文件没有属性 size 或 width。