Shortcuts

Source code for caer.color.core

#    _____           ______  _____
#  / ____/    /\    |  ____ |  __ \
# | |        /  \   | |__   | |__) | Caer - Modern Computer Vision
# | |       / /\ \  |  __|  |  _  /  Languages: Python, C, C++, Cuda
# | |___   / ____ \ | |____ | | \ \  http://github.com/jasmcaus/caer
#  \_____\/_/    \_ \______ |_|  \_\

# Licensed under the MIT License <http://opensource.org/licenses/MIT>
# SPDX-License-Identifier: MIT
# Copyright (c) 2020-2021 The Caer Authors <http://github.com/jasmcaus>

from ..adorad import Tensor, to_tensor

from ._bgr import bgr2gray, bgr2hsv, bgr2lab, bgr2rgb, bgr2hls, bgr2yuv, bgr2luv
from ._rgb import rgb2gray, rgb2hsv, rgb2lab, rgb2bgr, rgb2hls, rgb2yuv, rgb2luv
from ._gray import gray2lab, gray2rgb, gray2hsv, gray2bgr, gray2hls, gray2yuv, gray2luv
from ._hsv import hsv2gray, hsv2rgb, hsv2lab, hsv2bgr, hsv2hls, hsv2yuv, hsv2luv
from ._hls import hls2gray, hls2rgb, hls2lab, hls2bgr, hls2hsv, hls2yuv, hls2luv
from ._lab import lab2gray, lab2rgb, lab2hsv, lab2bgr, lab2hls, lab2yuv, lab2luv
from ._yuv import yuv2bgr, yuv2rgb, yuv2hsv, yuv2hls, yuv2gray, yuv2lab, yuv2luv
from ._luv import luv2bgr, luv2rgb, luv2hsv, luv2hls, luv2gray, luv2lab, luv2yuv

__all__ = [
    'to_rgb',
    'to_bgr',
    'to_gray',
    'to_hsv',
    'to_hls',
    'to_lab',
    'to_yuv',
    'to_luv',
]


[docs]def to_rgb(tens) -> Tensor: r''' Converts any supported colorspace to RGB Args: tens (Tensor) Returns: Tensor ''' if not isinstance(tens, Tensor): raise TypeError('`tens` must be a caer.Tensor') # Convert to tensor _ = ( tens._nullprt() ) # raises a ValueError if we're dealing with a Foreign Tensor with illegal `.cspace` value cspace = tens.cspace # If 'null', we assume we have a brand new Tensor if cspace == 'null': print( 'Warning: Caer was unable to assign a colorspace for a foreign tensor. Sticking with "rgb". This issue will be' 'fixed in a future update.' ) # We assume that the tens is a BGR image im = bgr2rgb(tens) im = to_tensor(im, cspace='rgb') return im elif cspace == 'rgb': return tens elif cspace == 'bgr': im = bgr2rgb(tens) return to_tensor(im, cspace='rgb') elif cspace == 'gray': im = gray2rgb(tens) return to_tensor(im, cspace='rgb') elif cspace == 'hls': im = hls2rgb(tens) return to_tensor(im, cspace='rgb') elif cspace == 'hsv': im = hsv2rgb(tens) return to_tensor(im, cspace='rgb') elif cspace == 'lab': im = lab2rgb(tens) return to_tensor(im, cspace='rgb') elif cspace == 'yuv': im = yuv2rgb(tens) return to_tensor(im, cspace='rgb') elif cspace == 'luv': im = luv2rgb(tens) return to_tensor(im, cspace='rgb')
[docs]def to_bgr(tens) -> Tensor: r''' Converts any supported colorspace to BGR Args: tens (Tensor) Returns: Tensor ''' if not isinstance(tens, Tensor): raise TypeError('`tens` must be a caer.Tensor') # Convert to tensor _ = ( tens._nullprt() ) # raises a ValueError if we're dealing with a Foreign Tensor with illegal `.cspace` value cspace = tens.cspace if cspace == 'bgr': return tens elif cspace == 'gray': im = gray2bgr(tens) return to_tensor(im, cspace='bgr') elif cspace == 'rgb': im = rgb2bgr(tens) return to_tensor(im, cspace='bgr') elif cspace == 'hls': im = hls2bgr(tens) return to_tensor(im, cspace='bgr') elif cspace == 'hsv': im = hsv2bgr(tens) return to_tensor(im, cspace='bgr') elif cspace == 'lab': im = lab2bgr(tens) return to_tensor(im, cspace='bgr') elif cspace == 'yuv': im = yuv2bgr(tens) return to_tensor(im, cspace='bgr') elif cspace == 'luv': im = luv2bgr(tens) return to_tensor(im, cspace='bgr')
[docs]def to_gray(tens) -> Tensor: r''' Converts any supported colorspace to grayscale Args: tens (Tensor) Returns: Tensor ''' if not isinstance(tens, Tensor): raise TypeError('`tens` must be a caer.Tensor') # Convert to tensor _ = ( tens._nullprt() ) # raises a ValueError if we're dealing with a Foreign Tensor with illegal `.cspace` value cspace = tens.cspace if cspace == 'gray': return tens elif cspace == 'bgr': im = bgr2gray(tens) return to_tensor(im, cspace='gray') elif cspace == 'rgb': im = rgb2gray(tens) return to_tensor(im, cspace='gray') elif cspace == 'hls': im = hls2gray(tens) return to_tensor(im, cspace='gray') elif cspace == 'hsv': im = hsv2gray(tens) return to_tensor(im, cspace='gray') elif cspace == 'lab': im = lab2gray(tens) return to_tensor(im, cspace='gray') elif cspace == 'yuv': im = yuv2gray(tens) return to_tensor(im, cspace='gray') elif cspace == 'luv': im = luv2gray(tens) return to_tensor(im, cspace='gray')
[docs]def to_hsv(tens) -> Tensor: r''' Converts any supported colorspace to hsv Args: tens (Tensor) Returns: Tensor ''' if not isinstance(tens, Tensor): raise TypeError('`tens` must be a caer.Tensor') # Convert to tensor _ = ( tens._nullprt() ) # raises a ValueError if we're dealing with a Foreign Tensor with illegal `.cspace` value cspace = tens.cspace if cspace == 'hsv': return tens elif cspace == 'bgr': im = bgr2hsv(tens) return to_tensor(im, cspace='hsv') elif cspace == 'rgb': im = rgb2hsv(tens) return to_tensor(im, cspace='hsv') elif cspace == 'hls': im = hls2hsv(tens) return to_tensor(im, cspace='hsv') elif cspace == 'gray': im = gray2hsv(tens) return to_tensor(im, cspace='hsv') elif cspace == 'lab': im = lab2hsv(tens) return to_tensor(im, cspace='hsv') elif cspace == 'yuv': im = yuv2hsv(tens) return to_tensor(im, cspace='hsv') elif cspace == 'luv': im = luv2hsv(tens) return to_tensor(im, cspace='hsv')
[docs]def to_hls(tens) -> Tensor: r''' Converts any supported colorspace to HLS Args: tens (Tensor) Returns: Tensor ''' if not isinstance(tens, Tensor): raise TypeError('`tens` must be a caer.Tensor') # Convert to tensor _ = ( tens._nullprt() ) # raises a ValueError if we're dealing with a Foreign Tensor with illegal `.cspace` value cspace = tens.cspace if cspace == 'hls': return tens elif cspace == 'bgr': im = bgr2hls(tens) return to_tensor(im, cspace='hls') elif cspace == 'rgb': im = rgb2hls(tens) return to_tensor(im, cspace='hls') elif cspace == 'hsv': im = hsv2hls(tens) return to_tensor(im, cspace='hls') elif cspace == 'gray': im = gray2hls(tens) return to_tensor(im, cspace='hls') elif cspace == 'lab': im = lab2hls(tens) return to_tensor(im, cspace='hls') elif cspace == 'yuv': im = yuv2hls(tens) return to_tensor(im, cspace='hls') elif cspace == 'luv': im = luv2hls(tens) return to_tensor(im, cspace='hls')
[docs]def to_lab(tens) -> Tensor: r''' Converts any supported colorspace to LAB Args: tens (Tensor) Returns: Tensor ''' if not isinstance(tens, Tensor): raise TypeError('`tens` must be a caer.Tensor') # Convert to tensor _ = ( tens._nullprt() ) # raises a ValueError if we're dealing with a Foreign Tensor with illegal `.cspace` value cspace = tens.cspace if cspace == 'lab': return tens elif cspace == 'bgr': im = bgr2lab(tens) return to_tensor(im, cspace='lab') elif cspace == 'rgb': im = rgb2lab(tens) return to_tensor(im, cspace='lab') elif cspace == 'hsv': im = hsv2lab(tens) return to_tensor(im, cspace='lab') elif cspace == 'gray': im = gray2lab(tens) return to_tensor(im, cspace='lab') elif cspace == 'hls': im = hls2lab(tens) return to_tensor(im, cspace='lab') elif cspace == 'yuv': im = yuv2lab(tens) return to_tensor(im, cspace='lab') elif cspace == 'luv': im = luv2lab(tens) return to_tensor(im, cspace='lab')
def to_yuv(tens) -> Tensor: r''' Converts any supported colorspace to YUV Args: tens (Tensor) Returns: Tensor ''' if not isinstance(tens, Tensor): raise TypeError('`tens` must be a caer.Tensor') # Convert to tensor _ = ( tens._nullprt() ) # raises a ValueError if we're dealing with a Foreign Tensor with illegal `.cspace` value cspace = tens.cspace if cspace == 'yuv': return tens elif cspace == 'bgr': im = bgr2yuv(tens) return to_tensor(im, cspace='yuv') elif cspace == 'rgb': im = rgb2yuv(tens) return to_tensor(im, cspace='yuv') elif cspace == 'hsv': im = hsv2yuv(tens) return to_tensor(im, cspace='yuv') elif cspace == 'gray': im = gray2yuv(tens) return to_tensor(im, cspace='yuv') elif cspace == 'hls': im = hls2yuv(tens) return to_tensor(im, cspace='yuv') elif cspace == 'lab': im = lab2yuv(tens) return to_tensor(im, cspace='yuv') elif cspace == 'luv': im = luv2yuv(tens) return to_tensor(im, cspace='yuv') def to_luv(tens) -> Tensor: r''' Converts any supported colorspace to LUV Args: tens (Tensor) Returns: Tensor ''' if not isinstance(tens, Tensor): raise TypeError('`tens` must be a caer.Tensor') # Convert to tensor _ = ( tens._nullprt() ) # raises a ValueError if we're dealing with a Foreign Tensor with illegal `.cspace` value cspace = tens.cspace if cspace == 'luv': return tens elif cspace == 'bgr': im = bgr2luv(tens) return to_tensor(im, cspace='luv') elif cspace == 'rgb': im = rgb2luv(tens) return to_tensor(im, cspace='luv') elif cspace == 'hsv': im = hsv2luv(tens) return to_tensor(im, cspace='luv') elif cspace == 'gray': im = gray2luv(tens) return to_tensor(im, cspace='luv') elif cspace == 'hls': im = hls2luv(tens) return to_tensor(im, cspace='luv') elif cspace == 'lab': im = lab2luv(tens) return to_tensor(im, cspace='luv') elif cspace == 'yuv': im = yuv2luv(tens) return to_tensor(im, cspace='luv')
Read the Docs v: latest
Versions
latest
stable
Downloads
pdf
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.