Internal change

PiperOrigin-RevId: 432116215

official/projects/volumetric_models/configs/semantic_segmentation_3d.py

@@ -23,7 +23,7 @@ from official.modeling import hyperparams
 from official.modeling import optimization
 from official.projects.volumetric_models.configs import backbones
 from official.projects.volumetric_models.configs import decoders
-from official.vision.beta.configs import common
+from official.vision.configs import common
 
 
 @dataclasses.dataclass

official/projects/volumetric_models/dataloaders/segmentation_input_3d.py

@@ -16,8 +16,8 @@
 
 from typing import Any, Dict, Sequence, Tuple
 import tensorflow as tf
-from official.vision.beta.dataloaders import decoder
-from official.vision.beta.dataloaders import parser
+from official.vision.dataloaders import decoder
+from official.vision.dataloaders import parser
 
 
 class Decoder(decoder.Decoder):

official/projects/volumetric_models/dataloaders/segmentation_input_3d_test.py

@@ -20,7 +20,7 @@ from absl.testing import parameterized
 import tensorflow as tf
 
 from official.projects.volumetric_models.dataloaders import segmentation_input_3d
-from official.vision.beta.dataloaders import tfexample_utils
+from official.vision.dataloaders import tfexample_utils
 
 
 class InputReaderTest(parameterized.TestCase, tf.test.TestCase):

official/projects/volumetric_models/modeling/backbones/unet_3d.py

@@ -25,7 +25,7 @@ from typing import Any, Mapping, Sequence
 import tensorflow as tf
 from official.modeling import hyperparams
 from official.projects.volumetric_models.modeling import nn_blocks_3d
-from official.vision.beta.modeling.backbones import factory
+from official.vision.modeling.backbones import factory
 
 layers = tf.keras.layers
 

official/projects/volumetric_models/modeling/factory.py

@@ -21,8 +21,8 @@ import tensorflow as tf
 from official.modeling import hyperparams
 from official.projects.volumetric_models.modeling.decoders import factory as decoder_factory
 from official.projects.volumetric_models.modeling.heads import segmentation_heads_3d
-from official.vision.beta.modeling import segmentation_model
-from official.vision.beta.modeling.backbones import factory as backbone_factory
+from official.vision.modeling import segmentation_model
+from official.vision.modeling.backbones import factory as backbone_factory
 
 
 def build_segmentation_model_3d(

official/projects/volumetric_models/modeling/nn_blocks_3d.py

@@ -20,7 +20,7 @@ from typing import Sequence, Union
 import tensorflow as tf
 
 from official.modeling import tf_utils
-from official.vision.beta.modeling.layers import nn_layers
+from official.vision.modeling.layers import nn_layers
 
 
 @tf.keras.utils.register_keras_serializable(package='Vision')

official/projects/volumetric_models/modeling/segmentation_model_test.py

@@ -21,7 +21,7 @@ import tensorflow as tf
 from official.projects.volumetric_models.modeling import backbones
 from official.projects.volumetric_models.modeling import decoders
 from official.projects.volumetric_models.modeling.heads import segmentation_heads_3d
-from official.vision.beta.modeling import segmentation_model
+from official.vision.modeling import segmentation_model
 
 
 class SegmentationNetworkUNet3DTest(parameterized.TestCase, tf.test.TestCase):

official/projects/volumetric_models/serving/export_saved_model.py

@@ -43,7 +43,7 @@ from official.common import registry_imports  # pylint: disable=unused-import
 from official.core import exp_factory
 from official.modeling import hyperparams
 from official.projects.volumetric_models.serving import semantic_segmentation_3d
-from official.vision.beta.serving import export_saved_model_lib
+from official.vision.serving import export_saved_model_lib
 
 FLAGS = flags.FLAGS
 

official/projects/volumetric_models/serving/semantic_segmentation_3d.py

@@ -22,7 +22,7 @@ import tensorflow as tf
 from official.projects.volumetric_models.modeling import backbones
 from official.projects.volumetric_models.modeling import decoders
 from official.projects.volumetric_models.modeling import factory
-from official.vision.beta.serving import export_base
+from official.vision.serving import export_base
 
 
 class SegmentationModule(export_base.ExportModule):

official/projects/volumetric_models/tasks/semantic_segmentation_3d_test.py

@@ -30,7 +30,7 @@ from official.projects.volumetric_models.evaluation import segmentation_metrics
 from official.projects.volumetric_models.modeling import backbones
 from official.projects.volumetric_models.modeling import decoders
 from official.projects.volumetric_models.tasks import semantic_segmentation_3d as img_seg_task
-from official.vision.beta.dataloaders import tfexample_utils
+from official.vision.dataloaders import tfexample_utils
 
 
 class SemanticSegmentationTaskTest(tf.test.TestCase, parameterized.TestCase):

official/projects/volumetric_models/train.py

@@ -19,7 +19,7 @@ import gin  # pylint: disable=unused-import
 
 from official.common import flags as tfm_flags
 from official.projects.volumetric_models import registry_imports  # pylint: disable=unused-import
-from official.vision.beta import train
+from official.vision import train
 
 
 def main(_):

official/projects/volumetric_models/train_test.py

@@ -21,7 +21,7 @@ from absl import logging
 from absl.testing import flagsaver
 import tensorflow as tf
 from official.projects.volumetric_models import train as train_lib
-from official.vision.beta.dataloaders import tfexample_utils
+from official.vision.dataloaders import tfexample_utils
 
 FLAGS = flags.FLAGS