使用 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 图像库中的代码也欢迎使用。任何帮助将不胜感激。

谢谢

Answer 1

我假设您正在尝试将单通道图像转换为 3 通道灰度图像。您正在读取图像 img = cv2.imread('bw.jpg')，默认情况下，如果您没有将任何参数传递给 cv2.imread()，那么它会读取 3 通道图像，而不管图像中的原始通道数。您可以简单地删除行 cv2.cvtColor(img, cv2.COLOR_GRAY2RGB)，因为 img 已经是一个只有灰度信息的 3 通道图像。

但是，如果您误以为 OpenCV 具有将 RGB 颜色填充到灰度图像的功能，那么您可能使用了错误的库。您可以查看其他开源项目，例如 this，它使用深度学习为您的图像着色。

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)

使用 Python 将灰度图像转换为其原始颜色格式

Converting a Grayscale image to its original color format using Python

python

opencv

python-imaging-library

gray-code