Saturday, June 27, 2015

過慮顏色取得想判別的顏色

將圖像從彩色轉換為灰階,BGR<->Gray,BGR<->HSV
用HSV分離顏色
H:色相 0~179
S:飽和度 0~255
V:明度 0~255
 對於色彩轉換的方法,我們用兩個最廣泛的函式:
cv2.cvtColor():BGR轉換到HSV
cv2.inRange():只顯示設定範圍內的顏色

import cv2
import numpy as np
#create a VideoCapture object
cap = cv2.VideoCapture(0)

while(1):

    # Capture Video from Camera
    _, frame = cap.read()

    # Convert BGR to HSV
    hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)

    # define range of blue color in HSV
    lower_blue = np.array([110,50,50])
    upper_blue = np.array([130,255,255])

    # define range of green color in HSV 
    lower_green = np.array([50,100,100])
    upper_green = np.array([70,255,255])

    # define range of red color in HSV 
    lower_red = np.array([-10,100,100])
    upper_red = np.array([10,255,255])

    # Threshold the HSV image to get only blue colors
    mask_blue = cv2.inRange(hsv, lower_blue, upper_blue)
    # Threshold the HSV image to get only green colors
    mask_green = cv2.inRange(hsv, lower_green, upper_green)
    # Threshold the HSV image to get only red colors   
    mask_red = cv2.inRange(hsv, lower_red, upper_red)
    # Bitwise-AND mask and original image

    res_blue = cv2.bitwise_and(frame,frame, mask= mask_blue)
    res_green = cv2.bitwise_and(frame,frame, mask= mask_green)
    res_red = cv2.bitwise_and(frame,frame, mask= mask_red)
    cv2.imshow('frame',frame)
    cv2.imshow('mask_blue)',mask_blue)
    cv2.imshow('mask_green',mask_green)
    cv2.imshow('mask_red',mask_red)

    cv2.imshow('res_blue',res_blue)
    cv2.imshow('res_green',res_green)
    cv2.imshow('res_red',res_red)
    k = cv2.waitKey(5) & 0xFF
    if k == 27:
        break

cv2.destroyAllWindows()
參考資料:

No comments:

Post a Comment