Skip to content

GitLab

Unverified Commit e405f3c3 authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

Make rotate_image_tensor's results fully BC with v1 (#7271)

parent 8f198e53
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
torchvision/transforms/v2/functional/_geometry.py
View file @ e405f3c3
...@@ -832,8 +832,8 @@ def rotate_image_tensor( ...@@ -832,8 +832,8 @@ def rotate_image_tensor(
center_f = [0.0, 0.0] center_f = [0.0, 0.0]
if center is not None: if center is not None:
if expand: if expand:
# TODO: Do we actually want to warn, or just document this?
warnings.warn("The provided center argument has no effect on the result if expand is True") warnings.warn("The provided center argument has no effect on the result if expand is True")
else:
# Center values should be in pixel coordinates but translated such that (0, 0) corresponds to image center. # Center values should be in pixel coordinates but translated such that (0, 0) corresponds to image center.
center_f = [(c - s * 0.5) for c, s in zip(center, [width, height])] center_f = [(c - s * 0.5) for c, s in zip(center, [width, height])]
......
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