关于 AttributeError: ‘NoneType‘ object has no attribute ‘text‘ 的三种解决方法
今天训练模型时,报错:AttributeError: 'NoneType' object has no attribute 'text'经查阅,错因在于 .xml 文件,并总结了三种方法一、原因原因是标注文件 .xml的<object>没有<difficult>的标签。difficlut 表明这个待检测目标很难识别,有可能是虽然视觉上很清楚,但是没有上下文的话还是很难确认它
·
今天训练模型时,报错:AttributeError: 'NoneType' object has no attribute 'text'
经查阅,错因在于 .xml 文件,并总结了三种方法
一、原因
原因是标注文件 .xml的<object>没有<difficult>的标签。
difficlut 表明这个待检测目标很难识别,有可能是虽然视觉上很清楚,但是没有上下文的话还是很难确认它属于哪个分类;标为difficult的目标在测试成绩的评估中一般会被忽略
二、解决方法(方法二最高效)
方法一、在.xml标注文件中把没有difficlut标签的补上。
== > 一般数据集不在少数,一个一个加太耗时了
方法二、忽略这个标签,在代码里把difficlut置为0
(1)找到报错的地方
(2)发现 difficult 的值通过 object.find 找到,那么我们就添加一句判断,如果找不到(即处理 .xml文件 无 difficult标签 的情况),那就默认为0
if object.find('difficult'):
difficult = float(object.find('difficult').text)
else:
difficult = 0
修改后:
可以正常训练了
方法三、批量对数据集中xml文件做处理
import xml.etree.ElementTree as ET
from os import getcwd
sets=[('2007', 'train'), ('2007', 'val'), ('2007', 'test')]
classes = ["hat","person"]
def convert_annotation(year, image_id, list_file):
in_file = open('VOCdevkit/VOC%s/Annotations/%s.xml'%(year, image_id),encoding='utf-8')
tree=ET.parse(in_file)
root = tree.getroot()
a=1
i=a
for obj in root.iter('item'):
#difficult = obj.find('').text
cls = obj.find('name').text
#if cls not in classes or int(difficult)==1:
#continue
cls_id = classes.index(cls)
xmlbox = obj.find('bndbox')
b = (int(xmlbox.find('xmin').text), int(xmlbox.find('ymin').text), int(xmlbox.find('xmax').text), int(xmlbox.find('ymax').text))
#print(image_id)
list_file.write(" " + ",".join([str(a) for a in b]) + ',' + str(cls_id))
for obj in root.iter('object'):#找xml文档里的object
try:
difficult = obj.find('difficult').text#找object对里的difficult对 在不同格式里可能找不到
cls = obj.find('name').text
if cls not in classes or int(difficult)==1:
continue
cls_id = classes.index(cls)
xmlbox = obj.find('bndbox')
#print(image_id)
b = (int(xmlbox.find('xmin').text), int(xmlbox.find('ymin').text), int(xmlbox.find('xmax').text), int(xmlbox.find('ymax').text))
list_file.write(" " + ",".join([str(a) for a in b]) + ',' + str(cls_id))
except Exception:
print(i,end="")
wd = getcwd()
for year, image_set in sets:
image_ids = open('VOCdevkit/VOC%s/ImageSets/Main/%s.txt'%(year, image_set),encoding='utf-8').read().strip().split()
list_file = open('%s_%s.txt'%(year, image_set), 'w',encoding='utf-8')
for image_id in image_ids:
list_file.write('%s/VOCdevkit/VOC%s/JPEGImages/%s.jpg'%(wd, year, image_id))
convert_annotation(year, image_id, list_file)
list_file.write('\n')
list_file.close()
更多推荐
已为社区贡献1条内容
所有评论(0)