当前位置: 首页>>代码示例>>Python>>正文


Python string_int_label_map_pb2.StringIntLabelMap方法代码示例

本文整理汇总了Python中object_detection.protos.string_int_label_map_pb2.StringIntLabelMap方法的典型用法代码示例。如果您正苦于以下问题:Python string_int_label_map_pb2.StringIntLabelMap方法的具体用法?Python string_int_label_map_pb2.StringIntLabelMap怎么用?Python string_int_label_map_pb2.StringIntLabelMap使用的例子?那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在object_detection.protos.string_int_label_map_pb2的用法示例。


在下文中一共展示了string_int_label_map_pb2.StringIntLabelMap方法的9个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。

示例1: load_labelmap

# 需要导入模块: from object_detection.protos import string_int_label_map_pb2 [as 别名]
# 或者: from object_detection.protos.string_int_label_map_pb2 import StringIntLabelMap [as 别名]
def load_labelmap(path):
  """Loads label map proto.

  Args:
    path: path to StringIntLabelMap proto text file.
  Returns:
    a StringIntLabelMapProto
  """
  with tf.gfile.GFile(path, 'r') as fid:
    label_map_string = fid.read()
    label_map = string_int_label_map_pb2.StringIntLabelMap()
    try:
      text_format.Merge(label_map_string, label_map)
    except text_format.ParseError:
      label_map.ParseFromString(label_map_string)
  _validate_label_map(label_map)
  return label_map 
开发者ID:ringringyi,项目名称:DOTA_models,代码行数:19,代码来源:label_map_util.py

示例2: load_labelmap

# 需要导入模块: from object_detection.protos import string_int_label_map_pb2 [as 别名]
# 或者: from object_detection.protos.string_int_label_map_pb2 import StringIntLabelMap [as 别名]
def load_labelmap(path):
  """Loads label map proto.

  Args:
    path: path to StringIntLabelMap proto text file.
  Returns:
    a StringIntLabelMapProto
  """
  with tf.gfile.GFile(path, 'r') as fid:
    label_map_string = fid.read()
    label_map = string_int_label_map_pb2.StringIntLabelMap()
    try:
      text_format.Merge(label_map_string, label_map)
    except text_format.ParseError:
      label_map.ParseFromString(label_map_string)
  return label_map 
开发者ID:datitran,项目名称:object_detector_app,代码行数:18,代码来源:label_map_util.py

示例3: _load_labelmap

# 需要导入模块: from object_detection.protos import string_int_label_map_pb2 [as 别名]
# 或者: from object_detection.protos.string_int_label_map_pb2 import StringIntLabelMap [as 别名]
def _load_labelmap(labelmap_path):
  """Loads labelmap from the labelmap path.

  Args:
    labelmap_path: Path to the labelmap.

  Returns:
    A dictionary mapping class name to class numerical id.
  """

  label_map = string_int_label_map_pb2.StringIntLabelMap()
  with open(labelmap_path, 'r') as fid:
    label_map_string = fid.read()
    text_format.Merge(label_map_string, label_map)
  labelmap_dict = {}
  for item in label_map.item:
    labelmap_dict[item.name] = item.id
  return labelmap_dict 
开发者ID:ahmetozlu,项目名称:vehicle_counting_tensorflow,代码行数:20,代码来源:oid_vrd_challenge_evaluation.py

示例4: _load_labelmap

# 需要导入模块: from object_detection.protos import string_int_label_map_pb2 [as 别名]
# 或者: from object_detection.protos.string_int_label_map_pb2 import StringIntLabelMap [as 别名]
def _load_labelmap(labelmap_path):
  """Loads labelmap from the labelmap path.

  Args:
    labelmap_path: Path to the labelmap.

  Returns:
    A dictionary mapping class name to class numerical id
    A list with dictionaries, one dictionary per category.
  """

  label_map = string_int_label_map_pb2.StringIntLabelMap()
  with open(labelmap_path, 'r') as fid:
    label_map_string = fid.read()
    text_format.Merge(label_map_string, label_map)
  labelmap_dict = {}
  categories = []
  for item in label_map.item:
    labelmap_dict[item.name] = item.id
    categories.append({'id': item.id, 'name': item.name})
  return labelmap_dict, categories 
开发者ID:ahmetozlu,项目名称:vehicle_counting_tensorflow,代码行数:23,代码来源:oid_od_challenge_evaluation.py

示例5: create_categories_from_labelmap

# 需要导入模块: from object_detection.protos import string_int_label_map_pb2 [as 别名]
# 或者: from object_detection.protos.string_int_label_map_pb2 import StringIntLabelMap [as 别名]
def create_categories_from_labelmap(label_map_path, use_display_name=True):
  """Reads a label map and returns categories list compatible with eval.

  This function converts label map proto and returns a list of dicts, each of
  which  has the following keys:
    'id': an integer id uniquely identifying this category.
    'name': string representing category name e.g., 'cat', 'dog'.

  Args:
    label_map_path: Path to `StringIntLabelMap` proto text file.
    use_display_name: (boolean) choose whether to load 'display_name' field
      as category name.  If False or if the display_name field does not exist,
      uses 'name' field as category names instead.

  Returns:
    categories: a list of dictionaries representing all possible categories.
  """
  label_map = load_labelmap(label_map_path)
  max_num_classes = max(item.id for item in label_map.item)
  return convert_label_map_to_categories(label_map, max_num_classes,
                                         use_display_name) 
开发者ID:ahmetozlu,项目名称:vehicle_counting_tensorflow,代码行数:23,代码来源:label_map_util.py

示例6: create_category_index_from_labelmap

# 需要导入模块: from object_detection.protos import string_int_label_map_pb2 [as 别名]
# 或者: from object_detection.protos.string_int_label_map_pb2 import StringIntLabelMap [as 别名]
def create_category_index_from_labelmap(label_map_path, use_display_name=True):
  """Reads a label map and returns a category index.

  Args:
    label_map_path: Path to `StringIntLabelMap` proto text file.
    use_display_name: (boolean) choose whether to load 'display_name' field
      as category name.  If False or if the display_name field does not exist,
      uses 'name' field as category names instead.

  Returns:
    A category index, which is a dictionary that maps integer ids to dicts
    containing categories, e.g.
    {1: {'id': 1, 'name': 'dog'}, 2: {'id': 2, 'name': 'cat'}, ...}
  """
  categories = create_categories_from_labelmap(label_map_path, use_display_name)
  return create_category_index(categories) 
开发者ID:ahmetozlu,项目名称:vehicle_counting_tensorflow,代码行数:18,代码来源:label_map_util.py

示例7: _generate_label_map

# 需要导入模块: from object_detection.protos import string_int_label_map_pb2 [as 别名]
# 或者: from object_detection.protos.string_int_label_map_pb2 import StringIntLabelMap [as 别名]
def _generate_label_map(self, num_classes):
    label_map_proto = string_int_label_map_pb2.StringIntLabelMap()
    for i in range(1, num_classes + 1):
      item = label_map_proto.item.add()
      item.id = i
      item.name = 'label_' + str(i)
      item.display_name = str(i)
    return label_map_proto 
开发者ID:ringringyi,项目名称:DOTA_models,代码行数:10,代码来源:label_map_util_test.py

示例8: _validate_label_map

# 需要导入模块: from object_detection.protos import string_int_label_map_pb2 [as 别名]
# 或者: from object_detection.protos.string_int_label_map_pb2 import StringIntLabelMap [as 别名]
def _validate_label_map(label_map):
  """Checks if a label map is valid.

  Args:
    label_map: StringIntLabelMap to validate.

  Raises:
    ValueError: if label map is invalid.
  """
  for item in label_map.item:
    if item.id < 1:
      raise ValueError('Label map ids should be >= 1.') 
开发者ID:ringringyi,项目名称:DOTA_models,代码行数:14,代码来源:label_map_util.py

示例9: test_keep_categories_with_unique_id

# 需要导入模块: from object_detection.protos import string_int_label_map_pb2 [as 别名]
# 或者: from object_detection.protos.string_int_label_map_pb2 import StringIntLabelMap [as 别名]
def test_keep_categories_with_unique_id(self):
    label_map_proto = string_int_label_map_pb2.StringIntLabelMap()
    label_map_string = """
      item {
        id:2
        name:'cat'
      }
      item {
        id:1
        name:'child'
      }
      item {
        id:1
        name:'person'
      }
      item {
        id:1
        name:'n00007846'
      }
    """
    text_format.Merge(label_map_string, label_map_proto)
    categories = label_map_util.convert_label_map_to_categories(
        label_map_proto, max_num_classes=3)
    self.assertListEqual([{
        'id': 2,
        'name': u'cat'
    }, {
        'id': 1,
        'name': u'child'
    }], categories) 
开发者ID:datitran,项目名称:object_detector_app,代码行数:32,代码来源:label_map_util_test.py


注:本文中的object_detection.protos.string_int_label_map_pb2.StringIntLabelMap方法示例由纯净天空整理自Github/MSDocs等开源代码及文档管理平台,相关代码片段筛选自各路编程大神贡献的开源项目,源码版权归原作者所有,传播和使用请参考对应项目的License;未经允许,请勿转载。