View source on GitHub |
Builds Mask R-CNN model.
tfm.vision.factory.build_maskrcnn(
input_specs: tf.keras.layers.InputSpec,
model_config: tfm.vision.configs.maskrcnn.MaskRCNN
,
l2_regularizer: Optional[tf.keras.regularizers.Regularizer] = None,
backbone: Optional[tf.keras.Model] = None,
decoder: Optional[tf.keras.Model] = None
) -> tf.keras.Model