Move anchor generator to keras_cv

PiperOrigin-RevId: 331081952

official/vision/beta/ops/anchor.py

@@ -18,7 +18,6 @@ import collections
 # Import libraries
 import tensorflow as tf
 from official.vision import keras_cv
-from official.vision.beta.ops.experimental import anchor_generator
 from official.vision.detection.utils.object_detection import argmax_matcher
 from official.vision.detection.utils.object_detection import balanced_positive_negative_sampler
 from official.vision.detection.utils.object_detection import box_list
@@ -318,7 +317,7 @@ def build_anchor_generator(min_level, max_level, num_scales, aspect_ratios,
     stride = 2**level
     strides[level] = stride
     anchor_sizes[level] = anchor_size * stride
-  anchor_gen = anchor_generator.AnchorGenerator(
+  anchor_gen = keras_cv.ops.AnchorGenerator(
       anchor_sizes=anchor_sizes,
       scales=scales,
       aspect_ratios=aspect_ratios,

official/vision/keras_cv/__init__.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
-"""Keras-NLP package definition."""
+"""Keras-CV package definition."""
 # pylint: disable=wildcard-import
 from official.vision.keras_cv.losses import *
 from official.vision.keras_cv.ops import *

official/vision/keras_cv/ops/__init__.py

@@ -13,4 +13,5 @@
 # limitations under the License.
 # ==============================================================================
 """Keras-CV layers package definition."""
+from official.vision.keras_cv.ops.anchor_generator import AnchorGenerator
 from official.vision.keras_cv.ops.iou_similarity import IouSimilarity

official/vision/beta/ops/experimental/anchor_generator_test.py → official/vision/keras_cv/ops/anchor_generator_test.py

@@ -18,7 +18,7 @@ from absl.testing import parameterized
 import tensorflow as tf
 from tensorflow.python.distribute import combinations
 from tensorflow.python.distribute import strategy_combinations
-from official.vision.beta.ops.experimental import anchor_generator
+from official.vision.keras_cv.ops import anchor_generator
 
 
 class AnchorGeneratorTest(parameterized.TestCase, tf.test.TestCase):