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


Python faster_rcnn_box_coder.FasterRcnnBoxCoder方法代码示例

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


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

示例1: __init__

# 需要导入模块: from object_detection.box_coders import faster_rcnn_box_coder [as 别名]
# 或者: from object_detection.box_coders.faster_rcnn_box_coder import FasterRcnnBoxCoder [as 别名]
def __init__(self):
    similarity_calc = region_similarity_calculator.IouSimilarity()
    matcher = argmax_matcher.ArgMaxMatcher(
        matched_threshold=ssd_constants.MATCH_THRESHOLD,
        unmatched_threshold=ssd_constants.MATCH_THRESHOLD,
        negatives_lower_than_unmatched=True,
        force_match_for_each_row=True)

    box_coder = faster_rcnn_box_coder.FasterRcnnBoxCoder(
        scale_factors=ssd_constants.BOX_CODER_SCALES)

    self.default_boxes = DefaultBoxes()('ltrb')
    self.default_boxes = box_list.BoxList(
        tf.convert_to_tensor(self.default_boxes))
    self.assigner = target_assigner.TargetAssigner(
        similarity_calc, matcher, box_coder) 
开发者ID:tensorflow,项目名称:benchmarks,代码行数:18,代码来源:ssd_dataloader.py

示例2: test_get_correct_relative_codes_after_encoding

# 需要导入模块: from object_detection.box_coders import faster_rcnn_box_coder [as 别名]
# 或者: from object_detection.box_coders.faster_rcnn_box_coder import FasterRcnnBoxCoder [as 别名]
def test_get_correct_relative_codes_after_encoding(self):
    boxes = [[10.0, 10.0, 20.0, 15.0], [0.2, 0.1, 0.5, 0.4]]
    anchors = [[15.0, 12.0, 30.0, 18.0], [0.1, 0.0, 0.7, 0.9]]
    expected_rel_codes = [[-0.5, -0.416666, -0.405465, -0.182321],
                          [-0.083333, -0.222222, -0.693147, -1.098612]]
    boxes = box_list.BoxList(tf.constant(boxes))
    anchors = box_list.BoxList(tf.constant(anchors))
    coder = faster_rcnn_box_coder.FasterRcnnBoxCoder()
    rel_codes = coder.encode(boxes, anchors)
    with self.test_session() as sess:
      rel_codes_out, = sess.run([rel_codes])
      self.assertAllClose(rel_codes_out, expected_rel_codes) 
开发者ID:ringringyi,项目名称:DOTA_models,代码行数:14,代码来源:faster_rcnn_box_coder_test.py

示例3: test_get_correct_relative_codes_after_encoding_with_scaling

# 需要导入模块: from object_detection.box_coders import faster_rcnn_box_coder [as 别名]
# 或者: from object_detection.box_coders.faster_rcnn_box_coder import FasterRcnnBoxCoder [as 别名]
def test_get_correct_relative_codes_after_encoding_with_scaling(self):
    boxes = [[10.0, 10.0, 20.0, 15.0], [0.2, 0.1, 0.5, 0.4]]
    anchors = [[15.0, 12.0, 30.0, 18.0], [0.1, 0.0, 0.7, 0.9]]
    scale_factors = [2, 3, 4, 5]
    expected_rel_codes = [[-1., -1.25, -1.62186, -0.911608],
                          [-0.166667, -0.666667, -2.772588, -5.493062]]
    boxes = box_list.BoxList(tf.constant(boxes))
    anchors = box_list.BoxList(tf.constant(anchors))
    coder = faster_rcnn_box_coder.FasterRcnnBoxCoder(
        scale_factors=scale_factors)
    rel_codes = coder.encode(boxes, anchors)
    with self.test_session() as sess:
      rel_codes_out, = sess.run([rel_codes])
      self.assertAllClose(rel_codes_out, expected_rel_codes) 
开发者ID:ringringyi,项目名称:DOTA_models,代码行数:16,代码来源:faster_rcnn_box_coder_test.py

示例4: test_get_correct_boxes_after_decoding

# 需要导入模块: from object_detection.box_coders import faster_rcnn_box_coder [as 别名]
# 或者: from object_detection.box_coders.faster_rcnn_box_coder import FasterRcnnBoxCoder [as 别名]
def test_get_correct_boxes_after_decoding(self):
    anchors = [[15.0, 12.0, 30.0, 18.0], [0.1, 0.0, 0.7, 0.9]]
    rel_codes = [[-0.5, -0.416666, -0.405465, -0.182321],
                 [-0.083333, -0.222222, -0.693147, -1.098612]]
    expected_boxes = [[10.0, 10.0, 20.0, 15.0], [0.2, 0.1, 0.5, 0.4]]
    anchors = box_list.BoxList(tf.constant(anchors))
    coder = faster_rcnn_box_coder.FasterRcnnBoxCoder()
    boxes = coder.decode(rel_codes, anchors)
    with self.test_session() as sess:
      boxes_out, = sess.run([boxes.get()])
      self.assertAllClose(boxes_out, expected_boxes) 
开发者ID:ringringyi,项目名称:DOTA_models,代码行数:13,代码来源:faster_rcnn_box_coder_test.py

示例5: test_get_correct_boxes_after_decoding_with_scaling

# 需要导入模块: from object_detection.box_coders import faster_rcnn_box_coder [as 别名]
# 或者: from object_detection.box_coders.faster_rcnn_box_coder import FasterRcnnBoxCoder [as 别名]
def test_get_correct_boxes_after_decoding_with_scaling(self):
    anchors = [[15.0, 12.0, 30.0, 18.0], [0.1, 0.0, 0.7, 0.9]]
    rel_codes = [[-1., -1.25, -1.62186, -0.911608],
                 [-0.166667, -0.666667, -2.772588, -5.493062]]
    scale_factors = [2, 3, 4, 5]
    expected_boxes = [[10.0, 10.0, 20.0, 15.0], [0.2, 0.1, 0.5, 0.4]]
    anchors = box_list.BoxList(tf.constant(anchors))
    coder = faster_rcnn_box_coder.FasterRcnnBoxCoder(
        scale_factors=scale_factors)
    boxes = coder.decode(rel_codes, anchors)
    with self.test_session() as sess:
      boxes_out, = sess.run([boxes.get()])
      self.assertAllClose(boxes_out, expected_boxes) 
开发者ID:ringringyi,项目名称:DOTA_models,代码行数:15,代码来源:faster_rcnn_box_coder_test.py

示例6: test_very_small_Width_nan_after_encoding

# 需要导入模块: from object_detection.box_coders import faster_rcnn_box_coder [as 别名]
# 或者: from object_detection.box_coders.faster_rcnn_box_coder import FasterRcnnBoxCoder [as 别名]
def test_very_small_Width_nan_after_encoding(self):
    boxes = [[10.0, 10.0, 10.0000001, 20.0]]
    anchors = [[15.0, 12.0, 30.0, 18.0]]
    expected_rel_codes = [[-0.833333, 0., -21.128731, 0.510826]]
    boxes = box_list.BoxList(tf.constant(boxes))
    anchors = box_list.BoxList(tf.constant(anchors))
    coder = faster_rcnn_box_coder.FasterRcnnBoxCoder()
    rel_codes = coder.encode(boxes, anchors)
    with self.test_session() as sess:
      rel_codes_out, = sess.run([rel_codes])
      self.assertAllClose(rel_codes_out, expected_rel_codes) 
开发者ID:ringringyi,项目名称:DOTA_models,代码行数:13,代码来源:faster_rcnn_box_coder_test.py

示例7: test_build_faster_rcnn_box_coder_with_non_default_parameters

# 需要导入模块: from object_detection.box_coders import faster_rcnn_box_coder [as 别名]
# 或者: from object_detection.box_coders.faster_rcnn_box_coder import FasterRcnnBoxCoder [as 别名]
def test_build_faster_rcnn_box_coder_with_non_default_parameters(self):
    box_coder_text_proto = """
      faster_rcnn_box_coder {
        y_scale: 6.0
        x_scale: 3.0
        height_scale: 7.0
        width_scale: 8.0
      }
    """
    box_coder_proto = box_coder_pb2.BoxCoder()
    text_format.Merge(box_coder_text_proto, box_coder_proto)
    box_coder_object = box_coder_builder.build(box_coder_proto)
    self.assertTrue(isinstance(box_coder_object,
                               faster_rcnn_box_coder.FasterRcnnBoxCoder))
    self.assertEqual(box_coder_object._scale_factors, [6.0, 3.0, 7.0, 8.0]) 
开发者ID:ringringyi,项目名称:DOTA_models,代码行数:17,代码来源:box_coder_builder_test.py

示例8: build

# 需要导入模块: from object_detection.box_coders import faster_rcnn_box_coder [as 别名]
# 或者: from object_detection.box_coders.faster_rcnn_box_coder import FasterRcnnBoxCoder [as 别名]
def build(box_coder_config):
  """Builds a box coder object based on the box coder config.

  Args:
    box_coder_config: A box_coder.proto object containing the config for the
      desired box coder.

  Returns:
    BoxCoder based on the config.

  Raises:
    ValueError: On empty box coder proto.
  """
  if not isinstance(box_coder_config, box_coder_pb2.BoxCoder):
    raise ValueError('box_coder_config not of type box_coder_pb2.BoxCoder.')

  if box_coder_config.WhichOneof('box_coder_oneof') == 'faster_rcnn_box_coder':
    return faster_rcnn_box_coder.FasterRcnnBoxCoder(scale_factors=[
        box_coder_config.faster_rcnn_box_coder.y_scale,
        box_coder_config.faster_rcnn_box_coder.x_scale,
        box_coder_config.faster_rcnn_box_coder.height_scale,
        box_coder_config.faster_rcnn_box_coder.width_scale
    ])
  if (box_coder_config.WhichOneof('box_coder_oneof') ==
      'mean_stddev_box_coder'):
    return mean_stddev_box_coder.MeanStddevBoxCoder()
  if box_coder_config.WhichOneof('box_coder_oneof') == 'square_box_coder':
    return square_box_coder.SquareBoxCoder(scale_factors=[
        box_coder_config.square_box_coder.y_scale,
        box_coder_config.square_box_coder.x_scale,
        box_coder_config.square_box_coder.length_scale
    ])
  raise ValueError('Empty box coder.') 
开发者ID:ringringyi,项目名称:DOTA_models,代码行数:35,代码来源:box_coder_builder.py

示例9: test_build_faster_rcnn_box_coder_with_defaults

# 需要导入模块: from object_detection.box_coders import faster_rcnn_box_coder [as 别名]
# 或者: from object_detection.box_coders.faster_rcnn_box_coder import FasterRcnnBoxCoder [as 别名]
def test_build_faster_rcnn_box_coder_with_defaults(self):
    box_coder_text_proto = """
      faster_rcnn_box_coder {
      }
    """
    box_coder_proto = box_coder_pb2.BoxCoder()
    text_format.Merge(box_coder_text_proto, box_coder_proto)
    box_coder_object = box_coder_builder.build(box_coder_proto)
    self.assertIsInstance(box_coder_object,
                          faster_rcnn_box_coder.FasterRcnnBoxCoder)
    self.assertEqual(box_coder_object._scale_factors, [10.0, 10.0, 5.0, 5.0]) 
开发者ID:ahmetozlu,项目名称:vehicle_counting_tensorflow,代码行数:13,代码来源:box_coder_builder_test.py

示例10: test_build_faster_rcnn_box_coder_with_non_default_parameters

# 需要导入模块: from object_detection.box_coders import faster_rcnn_box_coder [as 别名]
# 或者: from object_detection.box_coders.faster_rcnn_box_coder import FasterRcnnBoxCoder [as 别名]
def test_build_faster_rcnn_box_coder_with_non_default_parameters(self):
    box_coder_text_proto = """
      faster_rcnn_box_coder {
        y_scale: 6.0
        x_scale: 3.0
        height_scale: 7.0
        width_scale: 8.0
      }
    """
    box_coder_proto = box_coder_pb2.BoxCoder()
    text_format.Merge(box_coder_text_proto, box_coder_proto)
    box_coder_object = box_coder_builder.build(box_coder_proto)
    self.assertIsInstance(box_coder_object,
                          faster_rcnn_box_coder.FasterRcnnBoxCoder)
    self.assertEqual(box_coder_object._scale_factors, [6.0, 3.0, 7.0, 8.0]) 
开发者ID:ahmetozlu,项目名称:vehicle_counting_tensorflow,代码行数:17,代码来源:box_coder_builder_test.py


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