Internal change

PiperOrigin-RevId: 425419954

Internal change
PiperOrigin-RevId: 425419954
8a9a607c · Yeqing Li · A. Unique TensorFlower · b7bb52f0 · 8a9a607c · 8a9a607c
Commit 8a9a607c authored Jan 31, 2022 by Yeqing Li Committed by A. Unique TensorFlower Jan 31, 2022
14 changed files
--- a/official/vision/modeling/layers/roi_aligner_test.py
+++ b/official/vision/modeling/layers/roi_aligner_test.py
--- a/official/vision/modeling/layers/roi_generator.py
+++ b/official/vision/modeling/layers/roi_generator.py
--- a/official/vision/modeling/layers/roi_generator_test.py
+++ b/official/vision/modeling/layers/roi_generator_test.py
--- a/official/vision/modeling/layers/roi_sampler.py
+++ b/official/vision/modeling/layers/roi_sampler.py
--- a/official/vision/modeling/layers/roi_sampler_test.py
+++ b/official/vision/modeling/layers/roi_sampler_test.py
--- a/official/vision/modeling/maskrcnn_model.py
+++ b/official/vision/modeling/maskrcnn_model.py
--- a/official/vision/modeling/maskrcnn_model_test.py
+++ b/official/vision/modeling/maskrcnn_model_test.py
--- a/official/vision/modeling/retinanet_model.py
+++ b/official/vision/modeling/retinanet_model.py
--- a/official/vision/modeling/retinanet_model_test.py
+++ b/official/vision/modeling/retinanet_model_test.py
--- a/official/vision/modeling/segmentation_model.py
+++ b/official/vision/modeling/segmentation_model.py
--- a/official/vision/modeling/segmentation_model_test.py
+++ b/official/vision/modeling/segmentation_model_test.py
--- a/official/vision/modeling/video_classification_model.py
+++ b/official/vision/modeling/video_classification_model.py
--- a/official/vision/modeling/video_classification_model_test.py
+++ b/official/vision/modeling/video_classification_model_test.py
--- a/official/vision/utils/object_detection/visualization_utils.py
+++ b/official/vision/utils/object_detection/visualization_utils.py
@@ -34,7 +34,7 @@ import PIL.ImageFont as ImageFont
 import six
 import tensorflow as tf
-from official.vision.beta.ops import box_ops
+from official.vision.ops import box_ops
 from official.vision.utils.object_detection import shape_utils
 _TITLE_LEFT_MARGIN = 10