opencv-python——图片的HSV和RGB滚动条阈值分析器

在很多时候，我们需要找到图片的特定通道的阈值，就需要通过调整参数来尝试和确定数值。如果通过拖动的滚条来调节阈值，就可以提高调试效率，缩短试错时间。

opencv-python的滚动条API

创建滑动条——cv2.createTrackbar()

cv2.createTrackbar(trackbarName, windowName, value, count, onChange)-> None

参数：
	trackbarName：滑动条名称
	windowName：所在窗口名称
	value：滑动条初始值
	count：滑动条最大值
	onChange：回调函数名称，每次滑动都会调用回调函数
返回值：无

获取滑动条的数值——cv2.getTrackbar()

cv2.getTrackbar(trackbarname, winname)-> retval

参数：
	trackbarname：滑动条名称
	winname：所在窗口名
返回值：滚动条所在位置的值

HSV高低阈值分析器

# -*- coding:utf-8 -*-

import cv2
import numpy as np

"""
功能：读取一张图片，显示出来，转化为HSV色彩空间
     并通过滑块调节HSV阈值，实时显示
"""

image = cv2.imread('opencv/task/greenScreen01.jpg') # 根据路径读取一张图片
cv2.imshow("BGR", image) # 显示图片

hsv_low = np.array([0, 0, 0])
hsv_high = np.array([0, 0, 0])

# 下面几个函数，写得有点冗余

def h_low(value):
    hsv_low[0] = value

def h_high(value):
    hsv_high[0] = value

def s_low(value):
    hsv_low[1] = value

def s_high(value):
    hsv_high[1] = value

def v_low(value):
    hsv_low[2] = value

def v_high(value):
    hsv_high[2] = value

cv2.namedWindow('image',cv2.WINDOW_AUTOSIZE)
# 可以自己设定初始值，最大值255不需要调节
cv2.createTrackbar('H low', 'image', 35, 255, h_low) 
cv2.createTrackbar('H high', 'image', 90, 255, h_high)
cv2.createTrackbar('S low', 'image', 43, 255, s_low)
cv2.createTrackbar('S high', 'image', 255, 255, s_high)
cv2.createTrackbar('V low', 'image', 35, 255, v_low)
cv2.createTrackbar('V high', 'image', 255, 255, v_high)

while True:
    dst = cv2.cvtColor(image, cv2.COLOR_BGR2HSV) # BGR转HSV
    dst = cv2.inRange(dst, hsv_low, hsv_high) # 通过HSV的高低阈值，提取图像部分区域
    cv2.imshow('dst', dst)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break
cv2.destroyAllWindows()

运行展示

图像HSV值调节器

import numpy as np
import cv2

# 回调函数，未使用
def nothing(x):
    pass

# 将BGR图像转化为HSV图像
win_img = "new"
win_img_old = "old"
pic = cv2.imread("194.jpg", cv2.IMREAD_UNCHANGED) # 自己想要分析的照片
pic1 = cv2.cvtColor(pic, cv2.COLOR_BGR2HSV)

# 显示原图像做对比
cv2.namedWindow(win_img_old, cv2.WINDOW_NORMAL) 
cv2.imshow(win_img_old, pic) 

# 新图像窗口
cv2.namedWindow(win_img, cv2.WINDOW_AUTOSIZE) 
#初始化滚动条
cv2.createTrackbar("H",win_img, 100, 150, nothing)
cv2.createTrackbar("S",win_img, 100, 150, nothing)
cv2.createTrackbar("V",win_img, 100, 150, nothing)

while True:
	# ESC按下退出
    if cv2.waitKey(10) == 27:
        print("finish adjust picture and quit")
        break

	# 读取滚动条现在的滚动条的HSV信息
    h_value = float(cv2.getTrackbarPos("H",win_img)/100)
    s_value = float(cv2.getTrackbarPos("S",win_img)/100)
    v_value = float(cv2.getTrackbarPos("V",win_img)/100)
	# 拆分、读入新数据后，重新合成调整后的图片
    H, S, V = cv2.split(pic)
    new_pic = cv2.merge([np.uint8(H*h_value) , np.uint8(S*s_value) , np.uint8(V*v_value)])
    cv2.imshow(win_img, new_pic)

cv2.destroyAllWindows()

运行展示

RGB颜色分析器

import cv2
import numpy as np




def nothing(x):
    pass



img = np.zeros((640,480,3),np.uint8)
cv2.namedWindow('img')

cv2.createTrackbar('R','img',0,255,nothing)
cv2.createTrackbar('G','img',0,255,nothing)
cv2.createTrackbar('B','img',0,255,nothing)

switch = '0:OFF \n1:ON'
cv2.createTrackbar(switch,'img',0,1,nothing)

while(1):
    cv2.imshow('img',img)
    k = cv2.waitKey(5)
    if k == ord('q'):
        break
    elif k == ord('s'):
        # 按 s 键保存并退出
        # 保存结果
        lsImg = lsImg * 255
        lsImg = lsImg.astype(np.uint8)
        cv2.imwrite("lsImg.jpg", lsImg)
        break

    r = cv2.getTrackbarPos('R', 'img')
    g = cv2.getTrackbarPos('G', 'img')
    b = cv2.getTrackbarPos('B', 'img')
    s = cv2.getTrackbarPos(switch, 'img')

    if s == 0:
        img[:] = 0
    else:
        img[:] = [b,g,r]

cv2.destroyAllWindows()

运行展示