Internal change

PiperOrigin-RevId: 411136341

official/benchmark/keras_imagenet_benchmark.py

@@ -29,7 +29,7 @@ import tensorflow as tf  # pylint: disable=g-bad-import-order
 from official.benchmark import benchmark_wrappers
 from official.benchmark import keras_benchmark
 from official.benchmark.models import resnet_imagenet_main
-from official.vision.image_classification import classifier_trainer
+from official.legacy.image_classification import classifier_trainer
 
 MIN_TOP_1_ACCURACY = 0.76
 MAX_TOP_1_ACCURACY = 0.77

official/benchmark/models/cifar_preprocessing.py

@@ -23,7 +23,7 @@ import os
 from absl import logging
 import tensorflow as tf
 
-from official.vision.image_classification.resnet import imagenet_preprocessing
+from official.legacy.image_classification.resnet import imagenet_preprocessing
 
 HEIGHT = 32
 WIDTH = 32

official/benchmark/models/resnet_cifar_main.py

@@ -28,9 +28,9 @@ from official.benchmark.models import cifar_preprocessing
 from official.benchmark.models import resnet_cifar_model
 from official.benchmark.models import synthetic_util
 from official.common import distribute_utils
+from official.legacy.image_classification.resnet import common
 from official.utils.flags import core as flags_core
 from official.utils.misc import keras_utils
-from official.vision.image_classification.resnet import common
 
 
 LR_SCHEDULE = [  # (multiplier, epoch to start) tuples

official/benchmark/models/resnet_imagenet_main.py

@@ -22,14 +22,14 @@ from absl import flags
 from absl import logging
 import tensorflow as tf
 from official.common import distribute_utils
+from official.legacy.image_classification import test_utils
+from official.legacy.image_classification.resnet import common
+from official.legacy.image_classification.resnet import imagenet_preprocessing
+from official.legacy.image_classification.resnet import resnet_model
 from official.modeling import performance
 from official.utils.flags import core as flags_core
 from official.utils.misc import keras_utils
 from official.utils.misc import model_helpers
-from official.vision.image_classification import test_utils
-from official.vision.image_classification.resnet import common
-from official.vision.image_classification.resnet import imagenet_preprocessing
-from official.vision.image_classification.resnet import resnet_model
 
 
 def _cluster_last_three_conv2d_layers(model):

official/benchmark/models/resnet_imagenet_test.py

@@ -23,8 +23,8 @@ import tensorflow as tf
 
 from tensorflow.python.eager import context
 from official.benchmark.models import resnet_imagenet_main
+from official.legacy.image_classification.resnet import imagenet_preprocessing
 from official.utils.testing import integration
-from official.vision.image_classification.resnet import imagenet_preprocessing
 
 
 @parameterized.parameters(

official/benchmark/models/resnet_imagenet_test_tpu.py

@@ -21,8 +21,8 @@ from __future__ import print_function
 from absl.testing import parameterized
 import tensorflow as tf
 from official.benchmark.models import resnet_imagenet_main
+from official.legacy.image_classification.resnet import imagenet_preprocessing
 from official.utils.testing import integration
-from official.vision.image_classification.resnet import imagenet_preprocessing
 
 
 class KerasImagenetTest(tf.test.TestCase, parameterized.TestCase):

official/benchmark/resnet_ctl_imagenet_benchmark.py

@@ -23,8 +23,8 @@ from absl import flags
 import tensorflow as tf
 
 from official.benchmark import owner_utils
-from official.vision.image_classification.resnet import common
-from official.vision.image_classification.resnet import resnet_ctl_imagenet_main
+from official.legacy.image_classification.resnet import common
+from official.legacy.image_classification.resnet import resnet_ctl_imagenet_main
 from official.benchmark.perfzero_benchmark import PerfZeroBenchmark
 from official.benchmark import benchmark_wrappers
 from official.utils.flags import core as flags_core