Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Added temporary fix for channels-last-like tensors for Resize op #4979

Open
wants to merge 2 commits into
base: main
Choose a base branch
from
Open

Added temporary fix for channels-last-like tensors for Resize op #4979

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
93522e4
Added temporary fix for channels-last-like tensors for Resize op
vfdev-5 Nov 23, 2021
14d63d9
Merge branch 'main' into fix-4880-slow-resize
datumbox Aug 22, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions torchvision/transforms/functional_tensor.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -455,6 +455,14 @@ def resize(

img, need_cast, need_squeeze, out_dtype = _cast_squeeze_in(img, [torch.float32, torch.float64])

# Temporary fix for channels-last-like tensor
# Context: https://github.com/pytorch/vision/issues/4880 and
# https://github.com/pytorch/pytorch/issues/68430
# For channels-first tensor .contiguous() is no-op
# For unsqueezed 3D channels-list tensor,
# we transform it to contiguous 4D channels-first tensor
img = img.contiguous()

# Define align_corners to avoid warnings
align_corners = False if interpolation in ["bilinear", "bicubic"] else None

Expand Down