Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "9ea6ac1b07bc351943a6a0c0011e8ccb4ebb9b96"
Unverified Commit 56fb8411 authored by vfdev's avatar vfdev Committed by GitHub
Browse files

[proto] Fixed issue in FixedSizeCrop (#6532) 


Co-authored-by: default avatarPhilip Meier <github.pmeier@posteo.de>
parent 15b97d4b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/prototype/transforms/_geometry.py
View file @ 56fb8411
...@@ -801,7 +801,7 @@ class FixedSizeCrop(Transform): ...@@ -801,7 +801,7 @@ class FixedSizeCrop(Transform):
if needs_crop and bounding_boxes is not None: if needs_crop and bounding_boxes is not None:
bounding_boxes = cast( 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 = features.BoundingBox.new_like(
bounding_boxes, bounding_boxes,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment