Internal change

PiperOrigin-RevId: 438398969

official/vision/beta/projects/centernet/modeling/layers/cn_nn_blocks.py

@@ -18,7 +18,7 @@ from typing import List, Optional
 
 import tensorflow as tf
 
-from official.vision.beta.modeling.layers import nn_blocks
+from official.vision.modeling.layers import nn_blocks
 
 
 def _apply_blocks(inputs, blocks):

official/vision/beta/projects/centernet/modeling/layers/cn_nn_blocks_test.py

@@ -21,8 +21,8 @@ from absl.testing import parameterized
 import numpy as np
 import tensorflow as tf
 
-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.layers import nn_blocks
 
 
 class HourglassBlockPyTorch(tf.keras.layers.Layer):

official/vision/beta/projects/centernet/modeling/layers/detection_generator.py

@@ -25,9 +25,9 @@ from typing import Any, Mapping
 
 import tensorflow as tf
 
-from official.vision.beta.ops import box_ops
 from official.vision.beta.projects.centernet.ops import loss_ops
 from official.vision.beta.projects.centernet.ops import nms_ops
+from official.vision.ops import box_ops
 
 
 class CenterNetDetectionGenerator(tf.keras.layers.Layer):

official/vision/beta/projects/centernet/utils/checkpoints/load_weights.py

@@ -14,11 +14,11 @@
 
 """Functions used to load the ODAPI CenterNet checkpoint."""
 
-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.beta.projects.centernet.utils.checkpoints import config_classes
 from official.vision.beta.projects.centernet.utils.checkpoints import config_data
+from official.vision.modeling.backbones import mobilenet
+from official.vision.modeling.layers import nn_blocks
 
 Conv2DBNCFG = config_classes.Conv2DBNCFG
 HeadConvCFG = config_classes.HeadConvCFG