Skip to content

Commit

Permalink
Fix image.shape type (keras-team#2305)
Browse files Browse the repository at this point in the history
Fixed image.shape issue for tensorflow backend
  • Loading branch information
sachinprasadhs authored and yuvraj-wale committed Feb 8, 2024
1 parent 4e5e76a commit 4b7132a
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 4b7132a

Please sign in to comment.