plenoptic.data.color_wheel

plenoptic.data.color_wheel(as_gray=False)

Return an example image of a color wheel.

Parameters:

as_gray (bool (default: False)) – Whether to load a single grayscale channel or 3 RGB channels.

Return type:

Tensor

Returns:

image – 4d tensor of the image.

Examples

>>> import plenoptic as po
>>> color_wheel = po.data.color_wheel()
>>> color_wheel.shape
torch.Size([1, 3, 600, 600])
>>> po.plot.imshow(color_wheel, as_rgb=True, zoom=0.5)
<PyrFigure size ...>

(png, hires.png, pdf)

>>> import plenoptic as po
>>> color_wheel_gray = po.data.color_wheel(as_gray=True)
>>> color_wheel_gray.shape
torch.Size([1, 1, 600, 600])
>>> po.plot.imshow(color_wheel_gray, zoom=0.5)
<PyrFigure size ...>

(png, hires.png, pdf)