使用Python將灰度圖像轉換為其原始顏色格式

Question

嗨，我目前正在嘗試使用python中的Open CV將灰度圖像轉換為其原始顏色格式。

import cv2

img = cv2.imread('bw.jpg')

img = cv2.cvtColor(img, cv2.COLOR_GRAY2RGB)


cv2.imwrite('gray_image.png',gray_image)

執行此操作將產生錯誤：

error: (-215) scn == 1 && (dcn == 3 || dcn == 4) in function cv::cvtColor

也歡迎使用Python Imaging Library中的代碼。 任何幫助將不勝感激。

謝謝

Answer 1

我假設您正在嘗試將單通道圖像轉換為3通道灰度圖像。 您將圖像讀取為img = cv2.imread('bw.jpg') ，默認情況下，如果未將任何參數傳遞給cv2.imread() ，則它將讀取3通道圖像，而與原始通道數cv2.imread()在圖像中。 您可以簡單地刪除cv2.cvtColor(img, cv2.COLOR_GRAY2RGB) ，因為img已經是一個3通道圖像，僅包含灰度信息。

但是，如果您認為OpenCV具有將RGB顏色填充到灰度圖像中的功能，那么您可能使用了錯誤的庫。 您可以簽出其他像這樣的開源項目，這些項目使用深度學習使圖像着色。

Answer 2

看到錯誤的地方，請查看內聯注釋。

import cv2

img = cv2.imread('bw.jpg')

x = img.shape

# check for color or gray-scale image type.
if x[3] == 3:

   print 'Got color image'
   # variable "gray_image" linked to result.
   gray_image = cv2.cvtColor(img, cv2.COLOR_GRAY2RGB)

   cv2.imwrite('gray_image.png',gray_image) # varname no longer img > gray_image.

else:
    print 'Got black/white, single channel image.'
    url = 'https://github.com//gustavla//autocolorize'
    print "Using ZdaR's posted solution from %s" % (url)