b152ed9c3252c386f43ff2c01ceb29f11bbcd105,research/object_detection/dataset_tools/create_ava_actions_tf_record.py,Ava,get_label_map,#Ava#Any#,307

Before Change

    with open(path, "r") as f:
      current_id = -1
      current_label = ""
      for line in f:
        if "item {" in line:
          current_id = -1
          current_label = ""
        if "name:" in line:
          first_quote = line.find(""") + 1
          second_quote = line.find(""", first_quote)
          assert second_quote > -1
          current_label = line[first_quote:second_quote]
        if "id:" in line:
          current_id = int(line.split()[1])
        if "}" in line:
          label_map[current_id] = bytes23(current_label)
    logging.info(label_map)
    assert len(label_map) == NUM_CLASSES
    return label_map

After Change

  def get_label_map(self, path):
    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:
      //label_map_util.load_labelmap()
      //label_map_str = f.read()
      //print(str(label_map_str))

Instances