-
Notifications
You must be signed in to change notification settings - Fork 7.1k
Current limitation on transforms #3224
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.
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
Comments
It would be good if Resize* ops allowed to specify interpolation backend and supported native PyTorch interpolate function (that could execute on GPU) - then could potentially execute on TorchScript |
@vadimkantorov Hi! Do you mean tensor&pil by interpolation backends? Currently I think Resize* on tensor is based on interpolation in pytorch, which has limited modes. |
PR to make |
The torchvision transforms now have 2 backends (PIL and Tensor), here are some functional mismatch between them and some may-be-useful features that neither of them support. Details are listed in transforms.py and functional.py.
Supported by PIL but not Tensor:
Crop with crop size larger than the original image. #3297 Solved by #3333
Supported by Tensor but not PIL:
Supported by neither:
Not supported by torchscript (mostly not possible given the current jit support):
It is just a draft, let me know if I forget anything. cc @vfdev-5 @datumbox
The text was updated successfully, but these errors were encountered: