Internal change

PiperOrigin-RevId: 432010568

official/projects/movinet/configs/movinet.py

@@ -28,9 +28,9 @@ import dataclasses
 from official.core import config_definitions as cfg
 from official.core import exp_factory
 from official.modeling import hyperparams
-from official.vision.beta.configs import backbones_3d
-from official.vision.beta.configs import common
-from official.vision.beta.configs import video_classification
+from official.vision.configs import backbones_3d
+from official.vision.configs import common
+from official.vision.configs import video_classification
 
 
 @dataclasses.dataclass

official/projects/movinet/configs/movinet_test.py

@@ -20,7 +20,7 @@ import tensorflow as tf
 from official.core import config_definitions as cfg
 from official.core import exp_factory
 from official.projects.movinet.configs import movinet
-from official.vision.beta.configs import video_classification as exp_cfg
+from official.vision.configs import video_classification as exp_cfg
 
 
 class MovinetConfigTest(tf.test.TestCase, parameterized.TestCase):

official/projects/movinet/modeling/movinet.py

@@ -25,7 +25,7 @@ import tensorflow as tf
 
 from official.modeling import hyperparams
 from official.projects.movinet.modeling import movinet_layers
-from official.vision.beta.modeling.backbones import factory
+from official.vision.modeling.backbones import factory
 
 # Defines a set of kernel sizes and stride sizes to simplify and shorten
 # architecture definitions for configs below.

official/projects/movinet/modeling/movinet_layers.py

@@ -23,7 +23,7 @@ from typing import Any, Mapping, Optional, Sequence, Tuple, 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
 
 # Default kernel weight decay that may be overridden
 KERNEL_WEIGHT_DECAY = 1.5e-5

official/projects/movinet/modeling/movinet_layers_test.py

@@ -19,7 +19,7 @@ from absl.testing import parameterized
 import tensorflow as tf
 
 from official.projects.movinet.modeling import movinet_layers
-from official.vision.beta.modeling.layers import nn_layers
+from official.vision.modeling.layers import nn_layers
 
 
 class MovinetLayersTest(parameterized.TestCase, tf.test.TestCase):

official/projects/movinet/modeling/movinet_model.py

@@ -23,8 +23,8 @@ import tensorflow as tf
 
 from official.projects.movinet.configs import movinet as cfg
 from official.projects.movinet.modeling import movinet_layers
-from official.vision.beta.modeling import backbones
-from official.vision.beta.modeling import factory_3d as model_factory
+from official.vision.modeling import backbones
+from official.vision.modeling import factory_3d as model_factory
 
 
 @tf.keras.utils.register_keras_serializable(package='Vision')

official/projects/movinet/tools/quantize_movinet.py

@@ -38,8 +38,8 @@ import numpy as np
 import tensorflow.compat.v2 as tf
 import tensorflow_hub as hub
 
-from official.vision.beta.configs import video_classification as video_classification_configs
-from official.vision.beta.tasks import video_classification
+from official.vision.configs import video_classification as video_classification_configs
+from official.vision.tasks import video_classification
 
 tf.enable_v2_behavior()
 

official/projects/movinet/train.py

@@ -34,9 +34,6 @@ from absl import app
 from absl import flags
 import gin
 
-# pylint: disable=unused-import
-from official.common import registry_imports
-# pylint: enable=unused-import
 from official.common import distribute_utils
 from official.common import flags as tfm_flags
 from official.core import task_factory
@@ -48,6 +45,7 @@ from official.modeling import performance
 # pylint: disable=unused-import
 from official.projects.movinet.modeling import movinet
 from official.projects.movinet.modeling import movinet_model
+from official.vision import registry_imports
 # pylint: enable=unused-import
 
 FLAGS = flags.FLAGS

official/projects/movinet/train_test.py

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