You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
32 lines
900 B
32 lines
900 B
2 years ago
|
# -*- coding: utf-8 -*-
|
||
|
|
"""
|
||
|
|
@File : colorSpaceService.py
|
||
|
|
@Author: csc
|
||
|
|
@Date : 2022/6/22
|
||
|
|
色彩空间
|
||
|
|
"""
|
||
|
|
import cv2
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
|
||
|
|
def getRGB(imgs, args=None) -> tuple:
|
||
|
|
"""
|
||
|
|
返回 RGB 三个通道对应的彩色图
|
||
|
|
:return: Tuple<img, img, img>
|
||
|
|
"""
|
||
|
|
b, g, r = cv2.split(imgs[0])
|
||
|
|
zeros = np.zeros(imgs[0].shape[:2], dtype="uint8")
|
||
|
|
return cv2.merge([zeros, zeros, r]), cv2.merge([zeros, g, zeros]), cv2.merge([b, zeros, zeros])
|
||
|
|
|
||
|
|
|
||
|
|
def getHSV(imgs, args=None) -> tuple:
|
||
|
|
"""
|
||
|
|
返回 HSV 三个通道对应的彩色图
|
||
|
|
:return: Tuple<img, img, img>
|
||
|
|
"""
|
||
|
|
hsv = cv2.cvtColor(imgs[0], cv2.COLOR_BGR2HSV)
|
||
|
|
h, s, v = cv2.split(hsv)
|
||
|
|
zeros = np.zeros(imgs[0].shape[:2], dtype="uint8")
|
||
|
|
# binary, full = cv2.threshold(zeros, -1, 255, cv2.THRESH_BINARY)
|
||
|
|
return cv2.merge([h, zeros, zeros]), cv2.merge([zeros, s, zeros]), cv2.merge([zeros, zeros, v])
|