generate_sequence_gt_mat.py 文件源码

def load_gt(xml_file):
    res = []
    with open(xml_file) as f:
        xml = dict(xmltodict.parse(f.read())['annotation'])
        try:
            obj = xml['object']
        except KeyError:
            print "xml {} has no objects.".format(xml_file)
            return np.asarray(res)
        if type(obj) is not list:
            boxes = [obj]
        else:
            boxes = obj
        for box in boxes:
            track_id = str(box['trackid'])
            bbox = map(int, [box['bndbox']['xmin'],
                             box['bndbox']['ymin'],
                             box['bndbox']['xmax'],
                             box['bndbox']['ymax'],
                             track_id])
            res.append(bbox)
    return np.asarray(res)