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))