[proto] Fixed issue in FixedSizeCrop (#6532)

Co-authored-by: Philip Meier <github.pmeier@posteo.de>

[proto] Fixed issue in FixedSizeCrop (#6532)
Co-authored-by: Philip Meier <github.pmeier@posteo.de>
56fb8411 · vfdev · GitHub · 15b97d4b · 56fb8411
Unverified Commit 56fb8411 authored Sep 01, 2022 by vfdev Committed by GitHub Sep 01, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchvision/prototype/transforms/_geometry.py torchvision/prototype/transforms/_geometry.py +1 -1

No files found.
--- a/torchvision/prototype/transforms/_geometry.py
+++ b/torchvision/prototype/transforms/_geometry.py
@@ -801,7 +801,7 @@ class FixedSizeCrop(Transform):

        if needs_crop and bounding_boxes is not None:
            bounding_boxes = cast(
-                features.BoundingBox, F.crop(bounding_boxes, top=top, left=left, height=height, width=width)
+                features.BoundingBox, F.crop(bounding_boxes, top=top, left=left, height=new_height, width=new_width)
            )
            bounding_boxes = features.BoundingBox.new_like(
                bounding_boxes,