def fast_rcnn_head_generator(params): """Generator function for Fast R-CNN head architecture.""" return heads.FastrcnnHead(params.num_classes, params.num_convs, params.num_filters, params.use_separable_conv, params.num_fcs, params.fc_dims, params.use_batch_norm, batch_norm_relu=batch_norm_relu_generator( params.batch_norm))
def fast_rcnn_head_generator(params): """Generator function for Fast R-CNN head architecture.""" head_params = params.frcnn_head return heads.FastrcnnHead(params.architecture.num_classes, head_params.num_convs, head_params.num_filters, head_params.use_separable_conv, head_params.num_fcs, head_params.fc_dims, params.norm_activation.activation, head_params.use_batch_norm, norm_activation=norm_activation_generator( params.norm_activation))
def fast_rcnn_head_generator(params): """Generator function for Fast R-CNN head architecture.""" return heads.FastrcnnHead(params.num_classes, params.fast_rcnn_mlp_head_dim, batch_norm_relu=batch_norm_relu_generator( params.batch_norm))