c987d27ce7922d08f5d50bf0280b102548fac2a0,research/object_detection/dataset_tools/create_ava_actions_tf_record.py,Ava,get_label_map,#Ava#Any#,464

Before Change

    Parses a label map into {integer:string} format.
    label_map = {}
    label_map = label_map_util.load_labelmap(path)
    print(label_map)
    label_map_dict = {}
    for item in label_map.item:
        label_map_dict[item.name] = item.label_id
    with open(path, "rb") as f:

After Change

  def get_label_map(self, path):
    Parses a label map into {integer:string} format.
    label_map_dict = label_map_util.get_label_map_dict(path)
    label_map_dict = {v: bytes(k, "utf8") for k, v in label_map_dict.items()}
    logging.info(label_map_dict)
    return label_map_dict

Instances