How to use the torchgeometry.color.BgrToRgb 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 
[[2., 2.],
[2., 2.]],
[[3., 3.],
[3., 3.]]]) # 3x2x2
expected = torch.tensor([[[3., 3.],
[3., 3.]],
[[2., 2.],
[2., 2.]],
[[1., 1.],
[1., 1.]]]) # 3x2x2
f = color.BgrToRgb()
assert_allclose(f(data), expected)[[2., 2.],
[2., 2.]],
[[1., 1.],
[1., 1.]]],
[[[3., 3.],
[3., 3.]],
[[2., 2.],
[2., 2.]],
[[1., 1.],
[1., 1.]]]]) # 2x3x2x2
f = color.BgrToRgb()
out = f(data)
assert_allclose(out, expected)def test_gradcheck(self):
# prepare input data
data = torch.tensor([[[1., 1.],
[1., 1.]],
[[2., 2.],
[2., 2.]],
[[3., 3.],
[3., 3.]]]) # 3x2x2
data = utils.tensor_to_gradcheck_var(data) # to var
assert gradcheck(color.BgrToRgb(), (data,),
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