Skip to content

GitLab

Commit c5df7268 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 475493697
parent 64ecc937
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 3 deletions
official/projects/volumetric_models/modeling/factory.py
View file @ c5df7268
...@@ -13,7 +13,7 @@ ...@@ -13,7 +13,7 @@
# limitations under the License. # limitations under the License.
"""Factory methods to build models.""" """Factory methods to build models."""
from typing import Sequence, Union
# Import libraries # Import libraries
import tensorflow as tf import tensorflow as tf
...@@ -26,9 +26,11 @@ from official.vision.modeling.backbones import factory as backbone_factory ...@@ -26,9 +26,11 @@ from official.vision.modeling.backbones import factory as backbone_factory
def build_segmentation_model_3d( def build_segmentation_model_3d(
input_specs: tf.keras.layers.InputSpec, input_specs: Union[tf.keras.layers.InputSpec,
Sequence[tf.keras.layers.InputSpec]],
model_config: hyperparams.Config, model_config: hyperparams.Config,
l2_regularizer: tf.keras.regularizers.Regularizer = None) -> tf.keras.Model: # pytype: disable=annotation-type-mismatch # typed-keras l2_regularizer: tf.keras.regularizers.Regularizer = None
) -> tf.keras.Model: # pytype: disable=annotation-type-mismatch # typed-keras
"""Builds Segmentation model.""" """Builds Segmentation model."""
norm_activation_config = model_config.norm_activation norm_activation_config = model_config.norm_activation
backbone = backbone_factory.build_backbone( backbone = backbone_factory.build_backbone(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment