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