Rotate an image using torchvision.transform for a tensor?

It looks like you could generate a random tensor, and then add it to the tensor you want to add jitter. If you want more of a random shifting type jitter, I wrote code for that a few days ago here: How to split a (C x H x W) tensor into tiles?

For the rotations, you could probably also use Torch.flip(), and Torch.diag() if the rotations are using 90 degree intervals.

1 Like