def rpn_head_generator(params):
"""Generator function for RPN head architecture."""
return heads.RpnHead(params.min_level,
params.max_level,
params.anchors_per_location,
batch_norm_relu=batch_norm_relu_generator(
params.batch_norm))
def rpn_head_generator(params):
"""Generator function for RPN head architecture."""
return heads.RpnHead(params.min_level,
params.max_level,
params.anchors_per_location,
params.num_convs,
params.num_filters,
params.use_separable_conv,
params.use_batch_norm,
batch_norm_relu=batch_norm_relu_generator(
params.batch_norm))
def rpn_head_generator(params):
"""Generator function for RPN head architecture."""
head_params = params.rpn_head
return heads.RpnHead(head_params.min_level,
head_params.max_level,
head_params.anchors_per_location,
head_params.num_convs,
head_params.num_filters,
head_params.use_separable_conv,
head_params.use_batch_norm,
batch_norm_activation=batch_norm_activation_generator(
params.batch_norm_activation))
def rpn_head_generator(params):
"""Generator function for RPN head architecture."""
head_params = params.rpn_head
if head_params.anchors_per_location:
logging.info('[Deprecation]: `rpn_head.anchors_per_location` '
'is no longer used.')
anchor_aspect_ratios = len(params.anchor.aspect_ratios)
anchor_num_scales = params.anchor.num_scales
anchors_per_location = anchor_aspect_ratios * anchor_num_scales
return heads.RpnHead(params.architecture.min_level,
params.architecture.max_level,
anchors_per_location,
head_params.num_convs,
head_params.num_filters,
head_params.use_separable_conv,
params.batch_norm_activation.activation,
head_params.use_batch_norm,
batch_norm_activation=batch_norm_activation_generator(
params.batch_norm_activation))
