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

Args
model_type	string name of model type. It should be consistent with ModelConfig.model_type.
input_specs	tf.keras.layers.InputSpec.
model_config	a OneOfConfig. Model config.
num_classes	number of classes.
l2_regularizer	tf.keras.regularizers.Regularizer instance. Default to None.

Returns
tf.keras.Model instance of the backbone.