本文整理汇总了Python中object_detection.protos.region_similarity_calculator_pb2.RegionSimilarityCalculator方法的典型用法代码示例。如果您正苦于以下问题:Python region_similarity_calculator_pb2.RegionSimilarityCalculator方法的具体用法?Python region_similarity_calculator_pb2.RegionSimilarityCalculator怎么用?Python region_similarity_calculator_pb2.RegionSimilarityCalculator使用的例子?那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类object_detection.protos.region_similarity_calculator_pb2
的用法示例。
在下文中一共展示了region_similarity_calculator_pb2.RegionSimilarityCalculator方法的5个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: build
# 需要导入模块: from object_detection.protos import region_similarity_calculator_pb2 [as 别名]
# 或者: from object_detection.protos.region_similarity_calculator_pb2 import RegionSimilarityCalculator [as 别名]
def build(region_similarity_calculator_config):
"""Builds region similarity calculator based on the configuration.
Builds one of [IouSimilarity, IoaSimilarity, NegSqDistSimilarity] objects. See
core/region_similarity_calculator.proto for details.
Args:
region_similarity_calculator_config: RegionSimilarityCalculator
configuration proto.
Returns:
region_similarity_calculator: RegionSimilarityCalculator object.
Raises:
ValueError: On unknown region similarity calculator.
"""
if not isinstance(
region_similarity_calculator_config,
region_similarity_calculator_pb2.RegionSimilarityCalculator):
raise ValueError(
'region_similarity_calculator_config not of type '
'region_similarity_calculator_pb2.RegionsSimilarityCalculator')
similarity_calculator = region_similarity_calculator_config.WhichOneof(
'region_similarity')
if similarity_calculator == 'iou_similarity':
return region_similarity_calculator.IouSimilarity()
if similarity_calculator == 'ioa_similarity':
return region_similarity_calculator.IoaSimilarity()
if similarity_calculator == 'neg_sq_dist_similarity':
return region_similarity_calculator.NegSqDistSimilarity()
raise ValueError('Unknown region similarity calculator.')
示例2: testBuildIoaSimilarityCalculator
# 需要导入模块: from object_detection.protos import region_similarity_calculator_pb2 [as 别名]
# 或者: from object_detection.protos.region_similarity_calculator_pb2 import RegionSimilarityCalculator [as 别名]
def testBuildIoaSimilarityCalculator(self):
similarity_calc_text_proto = """
ioa_similarity {
}
"""
similarity_calc_proto = sim_calc_pb2.RegionSimilarityCalculator()
text_format.Merge(similarity_calc_text_proto, similarity_calc_proto)
similarity_calc = region_similarity_calculator_builder.build(
similarity_calc_proto)
self.assertTrue(isinstance(similarity_calc,
region_similarity_calculator.IoaSimilarity))
示例3: testBuildIouSimilarityCalculator
# 需要导入模块: from object_detection.protos import region_similarity_calculator_pb2 [as 别名]
# 或者: from object_detection.protos.region_similarity_calculator_pb2 import RegionSimilarityCalculator [as 别名]
def testBuildIouSimilarityCalculator(self):
similarity_calc_text_proto = """
iou_similarity {
}
"""
similarity_calc_proto = sim_calc_pb2.RegionSimilarityCalculator()
text_format.Merge(similarity_calc_text_proto, similarity_calc_proto)
similarity_calc = region_similarity_calculator_builder.build(
similarity_calc_proto)
self.assertTrue(isinstance(similarity_calc,
region_similarity_calculator.IouSimilarity))
示例4: testBuildNegSqDistSimilarityCalculator
# 需要导入模块: from object_detection.protos import region_similarity_calculator_pb2 [as 别名]
# 或者: from object_detection.protos.region_similarity_calculator_pb2 import RegionSimilarityCalculator [as 别名]
def testBuildNegSqDistSimilarityCalculator(self):
similarity_calc_text_proto = """
neg_sq_dist_similarity {
}
"""
similarity_calc_proto = sim_calc_pb2.RegionSimilarityCalculator()
text_format.Merge(similarity_calc_text_proto, similarity_calc_proto)
similarity_calc = region_similarity_calculator_builder.build(
similarity_calc_proto)
self.assertTrue(isinstance(similarity_calc,
region_similarity_calculator.
NegSqDistSimilarity))
示例5: build
# 需要导入模块: from object_detection.protos import region_similarity_calculator_pb2 [as 别名]
# 或者: from object_detection.protos.region_similarity_calculator_pb2 import RegionSimilarityCalculator [as 别名]
def build(region_similarity_calculator_config):
"""Builds region similarity calculator based on the configuration.
Builds one of [IouSimilarity, IoaSimilarity, NegSqDistSimilarity] objects. See
core/region_similarity_calculator.proto for details.
Args:
region_similarity_calculator_config: RegionSimilarityCalculator
configuration proto.
Returns:
region_similarity_calculator: RegionSimilarityCalculator object.
Raises:
ValueError: On unknown region similarity calculator.
"""
if not isinstance(
region_similarity_calculator_config,
region_similarity_calculator_pb2.RegionSimilarityCalculator):
raise ValueError(
'region_similarity_calculator_config not of type '
'region_similarity_calculator_pb2.RegionsSimilarityCalculator')
similarity_calculator = region_similarity_calculator_config.WhichOneof(
'region_similarity')
if similarity_calculator == 'iou_similarity':
return region_similarity_calculator.IouSimilarity()
if similarity_calculator == 'ioa_similarity':
return region_similarity_calculator.IoaSimilarity()
if similarity_calculator == 'neg_sq_dist_similarity':
return region_similarity_calculator.NegSqDistSimilarity()
if similarity_calculator == 'thresholded_iou_similarity':
return region_similarity_calculator.ThresholdedIouSimilarity(
region_similarity_calculator_config.thresholded_iou_similarity.threshold
)
raise ValueError('Unknown region similarity calculator.')
开发者ID:ahmetozlu,项目名称:vehicle_counting_tensorflow,代码行数:40,代码来源:region_similarity_calculator_builder.py