Internal change

PiperOrigin-RevId: 440464457
chemaalvarez · Apr 8, 2022 · f95f014 · f95f014
1 parent 663dfb6
commit f95f014
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/official/projects/qat/vision/tasks/retinanet_test.py b/official/projects/qat/vision/tasks/retinanet_test.py
@@ -24,8 +24,8 @@
 from official.core import exp_factory
 from official.modeling import optimization
 from official.projects.qat.vision.tasks import retinanet
-from official.vision.beta.dataloaders import tfexample_utils
 from official.vision.configs import retinanet as exp_cfg
+from official.vision.dataloaders import tfexample_utils
 
 
 class RetinaNetTaskTest(parameterized.TestCase, tf.test.TestCase):

diff --git a/official/vision/beta/projects/centernet/modeling/centernet_model_test.py b/official/vision/beta/projects/centernet/modeling/centernet_model_test.py
@@ -17,12 +17,12 @@
 from absl.testing import parameterized
 import tensorflow as tf
 
-from official.vision.beta.configs import common
 from official.vision.beta.projects.centernet.configs import backbones
 from official.vision.beta.projects.centernet.modeling import centernet_model
 from official.vision.beta.projects.centernet.modeling.backbones import hourglass
 from official.vision.beta.projects.centernet.modeling.heads import centernet_head
 from official.vision.beta.projects.centernet.modeling.layers import detection_generator
+from official.vision.configs import common
 
 
 class CenterNetTest(parameterized.TestCase, tf.test.TestCase):

diff --git a/official/vision/beta/projects/centernet/tasks/centernet.py b/official/vision/beta/projects/centernet/tasks/centernet.py
@@ -32,7 +32,7 @@
 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.dataloaders import tf_example_label_map_decoder
 from official.vision.evaluation import coco_evaluator
 from official.vision.modeling.backbones import factory
 

diff --git a/official/vision/beta/projects/yolo/tasks/yolo.py b/official/vision/beta/projects/yolo/tasks/yolo.py
@@ -36,7 +36,7 @@
 from official.vision.beta.projects.yolo.ops import preprocessing_ops
 from official.vision.beta.projects.yolo.tasks import task_utils
 from official.vision.dataloaders import tfds_factory
-from official.vision.dataloaders.google import tf_example_label_map_decoder
+from official.vision.dataloaders import tf_example_label_map_decoder
 from official.vision.evaluation import coco_evaluator
 from official.vision.ops import box_ops