03-python opencv提取彩色影象三通道

概述

本節實現的是提取出彩色影象的三個通道。

• 開啟一個彩色圖片
• 利用系統函式分離三通道
• 自行分離三通道
• 顯示顯示三通道圖片

實現過程

引用與開啟圖片

不再贅述，程式碼如下。

import cv2  
import numpy

img = cv2.imread("test.png")

系統函式分離三通道

opencv裡自帶了分離三通道的函式split()，返回值依次是藍色、綠色和紅色通道的灰度圖，程式碼如下：

b, g, r = cv2.split(img)
cv2.imshow("Blue 1"
 
, b)
cv2.imshow("Green 1", g)
cv2.imshow("Red 1", r)

自行分離三通道函式

定義了三個函式來獲取三個通道的子矩陣。

獲取紅色通道：

def get_red(img):
    redImg = img[:,:,2]
    return redImg

獲取綠色通道：

def get_green(img):
    greenImg = img[:,:,1]
    return greenImg

獲取藍色通道

def get_blue(img):
    blueImg = img[:,:,0]
    return
 
 blueImg

呼叫函式分離通道並顯示

呼叫上述三個函式，顯示分離結果，並與之前的比較。

b = get_blue(img)
g = get_green(img)
r = get_red(img)
cv2.imshow("Blue 2", b)
cv2.imshow("Green 2", g)
cv2.imshow("Red 2", r)

等待鍵盤輸入並關閉所有視窗

這裡cv2.waitKey裡的引數0表示等待輸入任何按鍵，當用戶輸入任何一個按鍵後即呼叫destroyAllWindows()關閉所有影象視窗。

cv2.waitKey (0)  
cv2.destroyAllWindows() 
 

原始碼

整個程式的原始碼如下：

# created by Huang Lu
# 2016/8/22 23:02
# Depart. of EE, Tsinghua Univ.

import cv2
import numpy

def get_red(img):
    redImg = img[:,:,2]
    return redImg

def get_green(img):
    greenImg = img[:,:,1]
    return greenImg

def get_blue(img):
    blueImg = img[:,:,0]
    return blueImg

if __name__ == '__main__':
    img = cv2.imread("test.png")
    b, g, r = cv2.split(img)
    cv2.imshow("Blue 1", b)
    cv2.imshow("Green 1", g)
    cv2.imshow("Red 1", r)
    b = get_blue(img)
    g = get_green(img)
    r = get_red(img)
    cv2.imshow("Blue 2", b)
    cv2.imshow("Green 2", g)
    cv2.imshow("Red 2", r)
    cv2.waitKey(0)
    cv2.destroyAllWindows()

也可以參考我的GitHub上的，點選這裡。

執行結果

在命令列進入該源程式所在目錄後，執行python main.py後即可顯示結果，結果如下：

參考