Internal change

PiperOrigin-RevId: 437123122

official/projects/basnet/configs/basnet.py

@@ -20,7 +20,7 @@ from official.core import config_definitions as cfg
 from official.core import exp_factory
 from official.modeling import hyperparams
 from official.modeling import optimization
-from official.vision.beta.configs import common
+from official.vision.configs import common
 
 
 @dataclasses.dataclass

official/projects/basnet/serving/basnet.py

@@ -17,7 +17,7 @@
 import tensorflow as tf
 
 from official.projects.basnet.tasks import basnet
-from official.vision.beta.serving import semantic_segmentation
+from official.vision.serving import semantic_segmentation
 
 
 MEAN_RGB = (0.485 * 255, 0.456 * 255, 0.406 * 255)

official/projects/basnet/serving/export_saved_model.py

@@ -41,7 +41,7 @@ from absl import flags
 from official.core import exp_factory
 from official.modeling import hyperparams
 from official.projects.basnet.serving import basnet
-from official.vision.beta.serving import export_saved_model_lib
+from official.vision.serving import export_saved_model_lib
 
 FLAGS = flags.FLAGS
 

official/projects/basnet/tasks/basnet.py

@@ -27,7 +27,7 @@ from official.projects.basnet.evaluation import metrics as basnet_metrics
 from official.projects.basnet.losses import basnet_losses
 from official.projects.basnet.modeling import basnet_model
 from official.projects.basnet.modeling import refunet
-from official.vision.beta.dataloaders import segmentation_input
+from official.vision.dataloaders import segmentation_input
 
 
 def build_basnet_model(

official/projects/basnet/train.py

@@ -22,7 +22,7 @@ from official.projects.basnet.configs import basnet as basnet_cfg
 from official.projects.basnet.modeling import basnet_model
 from official.projects.basnet.modeling import refunet
 from official.projects.basnet.tasks import basnet as basenet_task
-from official.vision.beta import train
+from official.vision import train
 
 
 if __name__ == '__main__':

official/vision/beta/projects/centernet/common/registry_imports.py

@@ -15,7 +15,7 @@
 """All necessary imports for registration."""
 
 # pylint: disable=unused-import
-from official.common import registry_imports
+from official.vision import registry_imports
 from official.vision.beta.projects.centernet.configs import centernet
 from official.vision.beta.projects.centernet.modeling import centernet_model
 from official.vision.beta.projects.centernet.modeling.backbones import hourglass