def segmentation_head_generator(params):
"""Generator function for Segmentation head architecture."""
return heads.SegmentationHead(
params.num_classes,
params.level,
params.num_convs,
params.use_batch_norm,
batch_norm_relu=batch_norm_relu_generator(params.batch_norm))
def segmentation_head_generator(params):
"""Generator function for segmentation head architecture."""
head_params = params.segmentation_head
return heads.SegmentationHead(
head_params.num_classes,
head_params.level,
head_params.num_convs,
head_params.upsample_factor,
head_params.num_downsample_channels,
head_params.use_batch_norm,
batch_norm_activation=batch_norm_activation_generator(
params.batch_norm_activation))
def segmentation_head_generator(params):
"""Generator function for Segmentation head architecture."""
return heads.SegmentationHead(params.num_classes, params.level,
params.num_convs)