diff --git a/keras_cv/layers/object_detection/anchor_generator.py b/keras_cv/layers/object_detection/anchor_generator.py index 30dd421afd..effc125143 100644 --- a/keras_cv/layers/object_detection/anchor_generator.py +++ b/keras_cv/layers/object_detection/anchor_generator.py @@ -172,7 +172,7 @@ def __call__(self, image=None, image_shape=None): "Expected `image` to be a Tensor of rank 3. Got " f"image.shape.rank={len(image.shape)}" ) - image_shape = image.shape + image_shape = tuple(image.shape) results = {} for key, generator in self.anchor_generators.items():