How to use the torchgeometry.color 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_color.py View on Github 
def test_gradcheck(self):
batch_size, channels, height, width = 2, 3, 4, 5
img = torch.ones(batch_size, channels, height, width)
img = utils.tensor_to_gradcheck_var(img) # to var
assert gradcheck(color.rgb_to_grayscale, (img,), raise_exception=True)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