Open voldemortX opened 3 years ago
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. But I think it could potentially execute on GPU or by torchscript?
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