python opencv 提取彩色影象三通道
阿新 • • 發佈:2019-02-13
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
後即可顯示結果,結果如下: