使用cv.inRange实现颜色物体分割

转：原文链接https://www.zhangshengrong.com/p/w4N7D8ddXr/
图侵转删

import cv2 as cv
import numpy as np



def Tcallback(event):
    # 重新获取阈值
    H = cv.getTrackbarPos('H', 'show_pic')
    S = cv.getTrackbarPos('S', 'show_pic')
    V = cv.getTrackbarPos('V', 'show_pic')
    # 定义HSV中红色的范围
    lower_red = np.array([H, S, V])
    upper_red = np.array([10, 255, 255])
    # 设置HSV的阈值使得只取红色
    mask = cv.inRange(hsv, lower_red, upper_red)
    res = cv.bitwise_and(img, img, mask=mask)
    cv.imshow('show_pic', res)


# 读图
img = cv.imread('static/1.jpg', 1)
hsv = cv.cvtColor(img, cv.COLOR_BGR2HSV)

# 创建窗口
cv.namedWindow('show_pic')
cv.resizeWindow('show_pic', 640, 480)

# 设置回调
cv.createTrackbar('H', 'show_pic', 0, 179, Tcallback)
cv.createTrackbar('S', 'show_pic', 43, 255, Tcallback)
cv.createTrackbar('V', 'show_pic', 46, 255, Tcallback)  # 第二个参数必须和namedWindow的第一个参数一致

# 定义HSV中红色的范围
lower_red = np.array([0, 43, 46])  # 其实红色是两块区域，如果是要做分割，可以先求出两个mask最后合并
upper_red = np.array([10, 255, 255])  # 红色范围：[0, 43, 46]~[10,255,255]∪[156, 43, 46]~[180,255,255]。
# 设置HSV的阈值使得只取红色
mask = cv.inRange(hsv, lower_red, upper_red)
res = cv.bitwise_and(img, img, mask=mask)  # 想看黑白的就imshow(mask)
cv.imshow('show_pic', res)

if cv.waitKey(0) == ord('q'):
    cv.destroyAllWindows()

结果图