def _load_label_map(self, label_map_path: str):
"""Opens and parses a given shot classification label map into memory.
The standard AutoML Video .pbtxt label map is assumed:
item {name: "Cat" id: 0}
item {name: "Dog" id: 1}
Args:
label_map_path: String value for the file path of the label map.
"""
with open(label_map_path, 'r') as f:
self.label_map, _ = vot_utils.parse_label_map(f.read())
def main():
parser = argparse.ArgumentParser()
parser.add_argument('--label_map',
help='.pbtxt AutoML generated label map',
required=True)
parser.add_argument('--output',
help='output DeepStream compatible label map',
required=True)
args = parser.parse_args()
with open(args.label_map, 'r') as f:
label_map_dict, label_map_list = utils.parse_label_map(f.read())
# If there is no label for index 0, fill with 'unknown.'
# VOT skips index 0 labels, but VCN has index 0.
if 0 not in label_map_dict:
label_map_list = ['unknown'] + label_map_list
label_map_list = label_map_list + ['']
label_map_txt = '\n'.join(label_map_list)
with open(args.output, 'w') as f:
f.write(label_map_txt)
def _load_label_map(self, label_map_path):
with open(label_map_path, 'r') as f:
self.label_map, _ = vot_utils.parse_label_map(f.read())