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.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将灰度图像转换为其原始颜色格式

2 个答案: