Internal change

PiperOrigin-RevId: 433134139

official/vision/beta/projects/centernet/configs/backbones.py

@@ -17,7 +17,7 @@
 import dataclasses
 
 from official.modeling import hyperparams
-from official.vision.beta.configs import backbones
+from official.vision.configs import backbones
 
 
 @dataclasses.dataclass

official/vision/beta/projects/centernet/configs/centernet.py

@@ -17,12 +17,13 @@
 import dataclasses
 import os
 from typing import List, Optional, Tuple
+
 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.beta.projects.centernet.configs import backbones
+from official.vision.configs import common
 
 
 TfExampleDecoderLabelMap = common.TfExampleDecoderLabelMap

official/vision/beta/projects/centernet/dataloaders/centernet_input.py

@@ -18,13 +18,13 @@ from typing import Tuple
 
 import tensorflow as tf
 
-from official.vision.beta.dataloaders import parser
-from official.vision.beta.dataloaders import utils
-from official.vision.beta.ops import box_ops
-from official.vision.beta.ops import preprocess_ops
 from official.vision.beta.projects.centernet.ops import box_list
 from official.vision.beta.projects.centernet.ops import box_list_ops
 from official.vision.beta.projects.centernet.ops import preprocess_ops as cn_prep_ops
+from official.vision.dataloaders import parser
+from official.vision.dataloaders import utils
+from official.vision.ops import box_ops
+from official.vision.ops import preprocess_ops
 
 
 CHANNEL_MEANS = (104.01362025, 114.03422265, 119.9165958)

official/vision/beta/projects/centernet/modeling/backbones/hourglass.py

@@ -19,10 +19,10 @@ from typing import Optional
 import tensorflow as tf
 
 from official.modeling import hyperparams
-from official.vision.beta.modeling.backbones import factory
-from official.vision.beta.modeling.backbones import mobilenet
-from official.vision.beta.modeling.layers import nn_blocks
 from official.vision.beta.projects.centernet.modeling.layers import cn_nn_blocks
+from official.vision.modeling.backbones import factory
+from official.vision.modeling.backbones import mobilenet
+from official.vision.modeling.layers import nn_blocks
 
 HOURGLASS_SPECS = {
     10: {

official/vision/beta/projects/centernet/modeling/backbones/hourglass_test.py

@@ -18,10 +18,10 @@ from absl.testing import parameterized
 import numpy as np
 import tensorflow as tf
 
-from official.vision.beta.configs import common
 from official.vision.beta.projects.centernet.common import registry_imports  # pylint: disable=unused-import
 from official.vision.beta.projects.centernet.configs import backbones
 from official.vision.beta.projects.centernet.modeling.backbones import hourglass
+from official.vision.configs import common
 
 
 class HourglassTest(tf.test.TestCase, parameterized.TestCase):

official/vision/beta/projects/centernet/ops/box_list_ops.py

@@ -16,8 +16,8 @@
 
 import tensorflow as tf
 
-from official.vision.beta.ops import sampling_ops
 from official.vision.beta.projects.centernet.ops import box_list
+from official.vision.ops import sampling_ops
 
 
 def _copy_extra_fields(boxlist_to_copy_to, boxlist_to_copy_from):

official/vision/beta/projects/centernet/ops/loss_ops.py

@@ -16,7 +16,7 @@
 
 import tensorflow as tf
 
-from official.vision.beta.ops import sampling_ops
+from official.vision.ops import sampling_ops
 
 
 def _get_shape(tensor, num_dims):

official/vision/beta/projects/centernet/ops/target_assigner.py

@@ -18,7 +18,7 @@ from typing import Dict, List
 
 import tensorflow as tf
 
-from official.vision.beta.ops import sampling_ops
+from official.vision.ops import sampling_ops
 
 
 def smallest_positive_root(a, b, c):

official/vision/beta/projects/centernet/ops/target_assigner_test.py

@@ -17,8 +17,8 @@
 from absl.testing import parameterized
 import tensorflow as tf
 
-from official.vision.beta.ops import preprocess_ops
 from official.vision.beta.projects.centernet.ops import target_assigner
+from official.vision.ops import preprocess_ops
 
 
 class TargetAssignerTest(tf.test.TestCase, parameterized.TestCase):

official/vision/beta/projects/centernet/tasks/centernet.py

@@ -22,11 +22,6 @@ import tensorflow as tf
 from official.core import base_task
 from official.core import input_reader
 from official.core import task_factory
-from official.vision.beta.dataloaders import tf_example_decoder
-from official.vision.beta.dataloaders import tfds_factory
-from official.vision.beta.dataloaders import tf_example_label_map_decoder
-from official.vision.beta.evaluation import coco_evaluator
-from official.vision.beta.modeling.backbones import factory
 from official.vision.beta.projects.centernet.configs import centernet as exp_cfg
 from official.vision.beta.projects.centernet.dataloaders import centernet_input
 from official.vision.beta.projects.centernet.losses import centernet_losses
@@ -35,6 +30,11 @@ from official.vision.beta.projects.centernet.modeling.heads import centernet_hea
 from official.vision.beta.projects.centernet.modeling.layers import detection_generator
 from official.vision.beta.projects.centernet.ops import loss_ops
 from official.vision.beta.projects.centernet.ops import target_assigner
+from official.vision.dataloaders import tf_example_decoder
+from official.vision.dataloaders import tfds_factory
+from official.vision.dataloaders.google import tf_example_label_map_decoder
+from official.vision.evaluation import coco_evaluator
+from official.vision.modeling.backbones import factory
 
 
 @task_factory.register_task_cls(exp_cfg.CenterNetTask)

official/vision/beta/projects/centernet/utils/tf2_centernet_checkpoint_converter.py

@@ -19,7 +19,6 @@ from absl import flags
 from absl import logging
 import tensorflow as tf
 
-from official.vision.beta.modeling.backbones import factory
 from official.vision.beta.projects.centernet.common import registry_imports  # pylint: disable=unused-import
 from official.vision.beta.projects.centernet.configs import backbones
 from official.vision.beta.projects.centernet.configs import centernet
@@ -28,6 +27,7 @@ from official.vision.beta.projects.centernet.modeling.heads import centernet_hea
 from official.vision.beta.projects.centernet.modeling.layers import detection_generator
 from official.vision.beta.projects.centernet.utils.checkpoints import load_weights
 from official.vision.beta.projects.centernet.utils.checkpoints import read_checkpoints
+from official.vision.modeling.backbones import factory
 
 FLAGS = flags.FLAGS