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())