Python SegmentationHead Examples

Example #1

File: factory.py Project: ydfq1019/tpu

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))

Example #2

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))

Example #3

File: factory.py Project: vitalyvels/tpu

def segmentation_head_generator(params):
    """Generator function for Segmentation head architecture."""
    return heads.SegmentationHead(params.num_classes, params.level,
                                  params.num_convs)