先看看效果:
接着上代码:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time : 2021/5/21 19:12
# @Author : 若谷
# @File : my_track_bar.py
# @Software: PyCharm
import numpy as np
import cv2
# 打开摄像头:0为本地摄像头,1为外接摄像头
cap = cv2.VideoCapture(1)
def nothing(x):
pass
# 给窗口命名
cv2.namedWindow('result')
cv2.resizeWindow('result',600,800)
# 设置hsv6个分量,分别为low_H,high_H,low_S,high_S,low_V,high_V
'''
cv2.createTrackbar参数解释:
第一个参数时滑动条的名字,
第二个参数是滑动条被放置的窗口的名字,
第三个参数是滑动条默认值,
第四个参数时滑动条的最大值,
第五个参数时回调函数,每次滑动都会调用回调函数。
'''
cv2.createTrackbar("Low_H", "result", 35, 255, nothing)
cv2.createTrackbar("High_H", "result", 70, 255, nothing)
cv2.createTrackbar("Low_S", "result", 40, 255, nothing)
cv2.createTrackbar("High_S", "result",200, 255, nothing)
cv2.createTrackbar("Low_V", "result", 100, 255, nothing)
cv2.createTrackbar("High_V", "result", 120, 255, nothing)
while (cap.isOpened()): # 循环读取每一帧
# 从摄像头读入
ret, frame = cap.read()
# 转换为HSV空间
hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
# 分别获取被拖动后的HSV值
'''
cv2.getTrackbarPos()参数解释:
第一个参数为滚动条的名称
第二个参数为窗口的名字
'''
low_h = cv2.getTrackbarPos('Low_H', 'result')
hign_h = cv2.getTrackbarPos('High_H', 'result')
low_s = cv2.getTrackbarPos('Low_S', 'result')
hign_s = cv2.getTrackbarPos('High_S', 'result')
low_v = cv2.getTrackbarPos('Low_V', 'result')
hign_v = cv2.getTrackbarPos('High_V', 'result')
# 分别设置高低阈值
low_threshold = np.array([low_h, low_s, low_v])
high_threshold = np.array([hign_h, hign_s, hign_v])
# 根据阈值构建掩膜
mask = cv2.inRange(hsv, low_threshold, high_threshold)
# 对原图像和掩膜进行位运算
res = cv2.bitwise_and(frame, frame, mask=mask)
cv2.imshow("frame", frame)
cv2.imshow("mask", mask)
cv2.imshow("res", res)
key = cv2.waitKey(1)
if key == 27: # Esc退出
break
# 释放摄像头
print('摄像头已关闭!')
cap.release()
cv2.destroyAllWindows()
写代码不易,亲,点个赞可好(* ̄︶ ̄)
图片版:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time : 2021/5/21 19:05
# @Author : 若谷
# @File : track_bar2.py
# @Software: PyCharm
import cv2
import numpy as np
def nothing(x):
pass
cv2.namedWindow('result')
cv2.resizeWindow('result',600,800)
# 设置hsv6个分量,分别为low_H,high_H,low_S,high_S,low_V,high_V
'''
cv2.createTrackbar参数解释:
第一个参数时滑动条的名字,
第二个参数是滑动条被放置的窗口的名字,
第三个参数是滑动条默认值,
第四个参数时滑动条的最大值,
第五个参数时回调函数,每次滑动都会调用回调函数。
'''
cv2.createTrackbar("Low_H", "result", 35, 255, nothing)
cv2.createTrackbar("High_H", "result", 70, 255, nothing)
cv2.createTrackbar("Low_S", "result", 40, 255, nothing)
cv2.createTrackbar("High_S", "result",200, 255, nothing)
cv2.createTrackbar("Low_V", "result", 100, 255, nothing)
cv2.createTrackbar("High_V", "result", 120, 255, nothing)
while True:
# img = cv2.imread('test.jpg')
img = cv2.imread('./images/car1.jpg')
resize = cv2.resize(img, None, fx=0.5, fy=0.5)
hsv = cv2.cvtColor(resize, cv2.COLOR_BGR2HSV)
# 分别获取被拖动后的HSV值
'''
cv2.getTrackbarPos()参数解释:
第一个参数为滚动条的名称
第二个参数为窗口的名字
'''
low_h = cv2.getTrackbarPos('Low_H', 'result')
hign_h = cv2.getTrackbarPos('High_H', 'result')
low_s = cv2.getTrackbarPos('Low_S', 'result')
hign_s = cv2.getTrackbarPos('High_S', 'result')
low_v = cv2.getTrackbarPos('Low_V', 'result')
hign_v = cv2.getTrackbarPos('High_V', 'result')
# 分别设置高低阈值
low_threshold = np.array([low_h, low_s, low_v])
high_threshold = np.array([hign_h, hign_s, hign_v])
# 根据阈值构建掩膜
mask = cv2.inRange(hsv, low_threshold, high_threshold)
# 对原图像和掩膜进行位运算
res = cv2.bitwise_and(resize, resize, mask=mask)
cv2.imshow("resize", resize)
cv2.imshow("mask", mask)
cv2.imshow("res", res)
if cv2.waitKey(1)== 27: # 按Esc退出,cv2.waitKey(x)里面的数不能为0
break
cv2.destroyAllWindows()
效果也是差不多的,就不放图的。
部分参考于:
1.https://www.cnpython.com/qa/642612