2014-02-06 15 views
23

我正在學習使用OpenCV進行實時應用程序的圖像處理。我對圖像做了一些閾值處理,並希望將輪廓標爲綠色,但它們並未呈現綠色,因爲我的圖像是黑白的。如何在OpenCV(Python)中將灰度圖像轉換爲RGB以便在二進制圖像中處理圖像之後可視化輪廓?

在程序的早期,我用gray = cv2.cvtColor(frame,cv2.COLOR_BGR2GRAY)將RGB從灰度轉換爲灰度,但回頭我很困惑,而函數backtorgb = cv2.cvtColor(gray,cv2。 CV_GRAY2RGB)給出AttributeError:'module'對象沒有屬性'CV_GRAY2RGB'。

下面的代碼似乎沒有繪製出綠色的輪廓 - 這是因爲它是灰度圖像?如果是這樣,我可以將灰度圖像轉換回RGB以綠色顯示輪廓?

import numpy as np 
import cv2 
import time 

cap = cv2.VideoCapture(0) 
while(cap.isOpened()): 

    ret, frame = cap.read() 

    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY) 

    ret, gb = cv2.threshold(gray,128,255,cv2.THRESH_BINARY) 

    gb = cv2.bitwise_not(gb) 

    contour,hier = cv2.findContours(gb,cv2.RETR_CCOMP,cv2.CHAIN_APPROX_SIMPLE) 

    for cnt in contour: 
     cv2.drawContours(gb,[cnt],0,255,-1) 
    gray = cv2.bitwise_not(gb) 

    cv2.drawContours(gray,contour,-1,(0,255,0),3) 

    cv2.imshow('test', gray) 

    if cv2.waitKey(1) & 0xFF == ord('q'): 
     break 

cap.release() 
cv2.destroyAllWindows() 
+3

你可以在原來的'框架'本身,而不是使用灰色圖像 –

+0

這就是我最終做的。感謝您的建議。 – user391339

回答

48

我宣傳我一個答案評論:

簡單的方法是:

You could draw in the original 'frame' itself instead of using gray image.

難的方法(方法你想實現):

backtorgb = cv2.cvtColor(gray,cv2.COLOR_GRAY2RGB)是正確的語法。

+0

我認爲這並不奏效,因爲你不能轉換回RGB。您將得到一個3槽陣列,但RGB值將與其在灰度圖像中的值相同。 – Megha

+2

是的。確實如此。但問題的作者對使用原始顏色不感興趣 –

+0

但是您是否真的看到了作者希望看到的綠色輪廓?即使在覆蓋COLOR_GRAY2RGB之後,我的圖像仍然顯示爲灰色。即使您將定義RGB通道,但我認爲最終結果仍然是灰色圖像 – Megha

3

一個你轉換你的圖像灰度級,你不能回來。你已經從三個頻道走到另一個,當你嘗試返回時,所有三個數字都是一樣的。所以簡短的回答是,你不能回去。原因你backtorgb功能此拋出這個錯誤是因爲它需要的格式爲:

CvtColor(input, output, CV_GRAY2BGR) 

OpenCV的使用BGR不是RGB,因此,如果您修復訂購它應該工作,但你的形象仍然是灰色的。

+0

實際上沒有工作。我在網上看過幾個參考文獻,表明上述可能,但不在我手中。我的pydev IDE的自動完成列出了CV_BGR2GRAY,但不是GRAY2BGR。困惑。謝謝! – user391339

+0

事實上,你的回答是真實的,我面臨同樣的問題 – Megha

7

試試這個:

import cv2 
import cv 

color_img = cv2.cvtColor(gray_img, cv.CV_GRAY2RGB) 

我發現,而CV模塊使用OpenCV的是,一些常量的CV2模塊中定義,等。

+0

或cv2.cv.CV_GRAY2RGB –

相關問題