本文整理匯總了Python中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使用的例子？那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類string_int_label_map_pb2的用法示例。
在下文中一共展示了string_int_label_map_pb2.StringIntLabelMap方法的5個代碼示例，這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚，您的評價將有助於係統推薦出更棒的Python代碼示例。
示例1: load_labelmap
# 需要導入模塊: import string_int_label_map_pb2 [as 別名]
# 或者: from 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 示例2: load_labelmap
# 需要導入模塊: import string_int_label_map_pb2 [as 別名]
# 或者: from 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 示例3: _validate_label_map
# 需要導入模塊: import string_int_label_map_pb2 [as 別名]
# 或者: from 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.') 示例4: create_category_index_from_labelmap
# 需要導入模塊: import string_int_label_map_pb2 [as 別名]
# 或者: from string_int_label_map_pb2 import StringIntLabelMap [as 別名]
def create_category_index_from_labelmap(label_map_path):
"""Reads a label map and returns a category index.
Args:
label_map_path: Path to `StringIntLabelMap` proto text file.
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'}, ...}
"""
label_map = load_labelmap(label_map_path)
max_num_classes = max(item.id for item in label_map.item)
categories = convert_label_map_to_categories(label_map, max_num_classes)
return create_category_index(categories) 示例5: _validate_label_map
# 需要導入模塊: import string_int_label_map_pb2 [as 別名]
# 或者: from 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 < 0:
raise ValueError('Label map ids should be >= 0.')
if (item.id == 0 and item.name != 'background' and
item.display_name != 'background'):
raise ValueError('Label map id 0 is reserved for the background label')