rotate_via_shear#

class deepinv.transform.rotate_via_shear(image, angle, center=None)[source]#

Bases:

2D rotation of image by angle via shear composition through FFT.

Parameters:

image (torch.Tensor) – input image of shape (B,C,H,W)
angle (torch.Tensor, float, int) – input rotation angles in degrees of shape (B,)

Returns:

torch.Tensor containing the rotated images of shape (B, C, H, W )