View source on GitHub |
Builds Segmentation model.
tfm.vision.factory.build_segmentation_model(
input_specs: tf.keras.layers.InputSpec,
model_config: tfm.vision.configs.semantic_segmentation.SemanticSegmentationModel
,
l2_regularizer: Optional[tf.keras.regularizers.Regularizer] = None,
backbone: Optional[tf.keras.Model] = None,
decoder: Optional[tf.keras.Model] = None
) -> tf.keras.Model