def shapeprior_head_generator(params): """Generator function for RetinaNet head architecture.""" return heads.ShapemaskPriorHead( params.num_classes, params.num_downsample_channels, params.mask_crop_size, params.use_category_for_mask, params.num_of_instances, params.min_mask_level, params.max_mask_level, params.num_clusters, params.temperature, params.shape_prior_path)
def shapeprior_head_generator(params): """Generator function for shape prior head architecture.""" head_params = params.shapemask_head return heads.ShapemaskPriorHead(params.architecture.num_classes, head_params.num_downsample_channels, head_params.mask_crop_size, head_params.use_category_for_mask, head_params.shape_prior_path)