How to use the torchgeometry.augmentation.RandomRotation function in torchgeometry
To help you get started, we’ve selected a few torchgeometry examples, based on popular ways it is used in public projects.
Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
kornia / kornia / test / test_augmentation.py View on Github 
def test_random_rotate_minmax_value(self):
transform = taug.RandomRotation(degrees=(-45, 0))
inp = torch.rand(1, 3, 4)
assert transform(inp).shape == (1, 3, 4)def test_random_rotate_minmax_value_batch(self):
transform = taug.RandomRotation(degrees=(-45, 0))
inp = torch.rand(2, 3, 3, 4)
assert transform(inp).shape == (2, 3, 3, 4)def test_random_rotate_value(self):
transform = taug.RandomRotation(degrees=90)
inp = torch.rand(1, 3, 4)
assert transform(inp).shape == (1, 3, 4)torchgeometry
differential geometric computer vision for deep learning
Package Health Score
63 / 100Popular torchgeometry functions
- torchgeometry.augmentation
- torchgeometry.augmentation.Grayscale
- torchgeometry.augmentation.RandomRotation
- torchgeometry.augmentation.Rotate
- torchgeometry.augmentation.ToTensor
- torchgeometry.color
- torchgeometry.color.bgr_to_rgb
- torchgeometry.color.BgrToRgb
- torchgeometry.color.Normalize
- torchgeometry.color.normalize
- torchgeometry.color.rgb_to_bgr
- torchgeometry.color.rgb_to_hsv
- torchgeometry.color.RgbToBgr
- torchgeometry.color.RgbToGrayscale
- torchgeometry.color.RgbToHsv
- torchgeometry.DepthWarper
- torchgeometry.utils
- torchgeometry.utils.create_pinhole
- torchgeometry.utils.tensor_to_image