The function in the corresponding libtorch torchvision.transforms.Compose
I think transforms
are still not completely ported and while some transformations already work on tensors instead of PIL.Image
s, I don’t think Compose
was ported yet.
This PR seems related and I’m sure contributions are welcome in case you would be interested to port some functions.
CC @fmassa to correct me in case I’m missing something.