def coarsemask_head_generator(params): """Generator function for RetinaNet head architecture.""" return heads.ShapemaskCoarsemaskHead(params.num_classes, params.num_downsample_channels, params.mask_crop_size, params.use_category_for_mask, params.num_convs)
def coarsemask_head_generator(params): """Generator function for ShapeMask coarse mask head architecture.""" head_params = params.shapemask_head return heads.ShapemaskCoarsemaskHead( params.architecture.num_classes, head_params.num_downsample_channels, head_params.mask_crop_size, head_params.use_category_for_mask, head_params.num_convs, batch_norm_activation=batch_norm_activation_generator( params.batch_norm_activation))