opencv 颜色空间转换

颜色空间转换

'''
在 OpenCV 中有超过150 种进行颜色空间转换的方法。但是你以后就会 发现我们经常用到的也就两种 BGR↔Gray 和 BGR↔HSV。
我们 用到的函数是 cv2.cvtColor(input_image flag) 其中 flag 就是转换类型。
对于 BGR↔Gray 的转换， 我们 使用的 flag 就是 cv2.COLOR_BGR2GRAY。 同样对于 BGR↔HSV 的转换 我们用的 flag 就是 cv2.COLOR_BGR2HSV
'''
import cv2
from pprint import pprint
flags = [i for i in dir(cv2) if i.startswith('COLOR_')]
pprint(flags)
#在 OpenCV 的 HSV 格式中 H 色彩/色度 的取值范围是 [0 179]
#  S 饱和度 的取值范围 [0 255]
# V 亮度 的取值范围 [0 255]。

物体跟踪

将一幅图片从BGR转换为HSV，因为HSV颜色空间中比BGR空间更容易表示一个特定颜色。

import cv2
import numpy as np

# 定蓝色的阈值
# lower = np.array([110, 50, 50])
# upper = np.array([130, 255, 255])
#黄色
lower = np.array([20, 100, 100])
upper = np.array([30, 255, 255])
# 黑色
# lower = np.array([0, 0, 0])
# upper = np.array([180, 255, 30])

img=cv2.imread('./img1.png')
# 换到 HSV
hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
# 根据阈值构建掩模
mask = cv2.inRange(hsv, lower, upper)
# mask = cv2.inRange(hsv, lower_black, upper_black)
# 对原图像和掩模位运算
res = cv2.bitwise_and(img, img, mask=mask)

# 显示图像
cv2.imshow('image', np.hstack((img,res)))
cv2.imwrite('image.png',np.hstack((img,res)))
k = cv2.waitKey()

掩膜

图像掩模是用选定的图像、图形或物体、对待处理的图像(全部或局部)进行遮挡来控制图像处理的区域或处理过程

 得到掩膜的过程。就是遍历图像的每一个像素，比较像素是不是在lower-upper之间，如果在那么则将这个像素置为255

import cv2
import numpy as np

lower = np.array([20, 100, 100])
upper = np.array([30, 255, 255])

img=cv2.imread('./img1.png')
# 换到 HSV
hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
# 根据阈值构建掩模
mask = cv2.inRange(hsv, lower, upper)

# print('mask',mask[40:48,40:49])
# print('hsv',hsv[40:48,40:49])
print(mask[45,45])
print(hsv[45,45])
cv2.imshow('image',mask)
cv2.waitKey()