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)