def mask_rcnn_head_generator(params): """Generator function for Mask R-CNN head architecture.""" return heads.MaskrcnnHead(params.num_classes, params.mask_target_size, params.use_batch_norm, batch_norm_relu=batch_norm_relu_generator( params.batch_norm))
def mask_rcnn_head_generator(params): """Generator function for Mask R-CNN head architecture.""" head_params = params.mrcnn_head return heads.MaskrcnnHead( head_params.num_classes, head_params.mask_target_size, 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))