使用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)