Skip to content

Commit

Permalink
Fix image.shape type (#2305)
Browse files Browse the repository at this point in the history
Fixed image.shape issue for tensorflow backend
  • Loading branch information
sachinprasadhs authored Jan 18, 2024
1 parent 5796b76 commit 01e5c34
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion keras_cv/layers/object_detection/anchor_generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -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():
Expand Down

0 comments on commit 01e5c34

Please sign in to comment.