- 函数通过遍历图像名称列表中的每个图像名称,从对应的XML标注文件中提取图像的标签信息。
- 对于每个图像,函数会解析对应的XML文件,找到其中的
<name>
标签,提取其中的类别名称信息。
def load_image_label_from_xml(img_name, PATH_TO_DATASET):
from xml.dom import minidom
elem_list = minidom.parse(os.path.join(PATH_TO_DATASET, ANNOT_FOLDER_NAME, decode_int_filename(img_name) + '.xml')).getElementsByTagName('name')
multi_cls_lab = np.zeros((N_CAT), np.float32)
for elem in elem_list:
cat_name = elem.firstChild.data
if cat_name in CAT_LIST:
cat_num = CAT_NAME_TO_NUM[cat_name]
multi_cls_lab[cat_num] = 1.0
return multi_cls_lab