How can I rotate random degree in 4D tensor (B,C,H,W)

vision
#1

How can I rotate random degree in 4D tensor (B,C,H,W)? Thanks

1 Like
#2

see here:
torchvision.transforms.functional.rotate

1 Like
#3

4D tensor, not PIL.Image.

#4

I typically use scipy’s rotate function:
https://docs.scipy.org/doc/scipy-0.16.1/reference/generated/scipy.ndimage.interpolation.rotate.html

Unfortunately it’s not very fast and you have to apply it on a numpy tensor, i.e. before you convert it to a pytorch tensor.

#5

What about this:

#6

The function torch.rot90(tensor, number_of_times_of_rotation:int, rotate_axis:list) is available