opencv-python提取图像中某种色彩的物体

提取带某种色彩的物体

• 获取视频中的每一帧
• 将色彩空间从BGR转换为HSV
• 为HSV图像设置所需提取色彩的阈值
• 根据阈值提取mask，再将原图像与mask进行位与运算
  

import cv2
import numpy as np

cap = cv2.VideoCapture(0)

while True:

    _, frame = cap.read()

    # 将色彩空间从BGR转为HSV
    hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)

    # 定义蓝色在HSV色彩空间中的范围
    lower_blue = np.array([110,50,50])
    upper_blue = np.array([130,255,255])

    # 根据阈值生成掩膜
    mask = cv2.inRange(hsv, lower_blue, upper_blue)

    # 将掩膜与原图像进行位与运算
    res = cv2.bitwise_and(frame, frame, mask=mask)

    cv2.imshow('frame',frame)
    cv2.imshow('mask',mask)
    cv2.imshow('res',res)
    k = cv2.waitKey(5) & 0xFF
    if k == 27:
        break

cv2.destroyAllWindows()