tfm.vision.factory_3d.build_model

Builds backbone from a config.

tfm.vision.factory_3d.build_model(
    model_type: str,
    input_specs: tf.keras.layers.InputSpec,
    model_config: tfm.hyperparams.Config,
    num_classes: int,
    l2_regularizer: tf.keras.regularizers.Regularizer = None
) -> tf.keras.Model

model_typestring name of model type. It should be consistent with ModelConfig.model_type.
input_specstf.keras.layers.InputSpec.
model_configa OneOfConfig. Model config.
num_classesnumber of classes.
l2_regularizertf.keras.regularizers.Regularizer instance. Default to None.

tf.keras.Model instance of the backbone.