Python OpenCV脚本中的HSV2BGR转换失败

Question

我的脚本应该采用灰度图像并将值映射到色调。

#!/usr/bin/env python

import cv2
import numpy

infile = cv2.imread('Lenna.png')

infile = infile[:,:,0]

hues = (numpy.array(infile)/255.)*179

outimageHSV = numpy.array([[[b,255,255] for b in a] for a in hues]).astype(int)

outimageBGR = cv2.cvtColor(outimageHSV, cv2.COLOR_HSV2BGR)

cv2.imshow('dst_rt', outimageBGR)
cv2.waitKey(0)
cv2.destroyAllWindows()

它在使用cvtColor的行上失败了，我得到了这个错误：

OpenCV Error: Assertion failed (depth == CV_8U || depth == CV_16U || depth == CV_32F) in cvtColor, file /tmp/opencv20150506-38415-u2kidu/opencv-2.4.11/modules/imgproc/src/color.cpp, line 3644
Traceback (most recent call last):
  File "luma2hue.py", line 16, in <module>
    outimageBGR = cv2.cvtColor(outimageHSV, cv2.COLOR_HSV2BGR)
cv2.error: /tmp/opencv20150506-38415-u2kidu/opencv-2.4.11/modules/imgproc/src/color.cpp:3644: error: (-215) depth == CV_8U || depth == CV_16U || depth == CV_32F in function cvtColor

我是否需要对我的outimageHSV数组执行其他操作以使其为cvtColor做好准备？

Answer 1

错误消息暗示cv2.cvtColor需要（颜色）深度为8或16位无符号整数（8U，16U）或32位浮点（32F）的图像。 尝试将astype(int)更改为astype(numpy.uint8)

Answer 2

outimageHSV需要作为uint8进行转换。

import numpy as np

outimageHSV = np.uint8(outimageHSV)
outimageBGR = cv2.cvtColor(outimageHSV, cv2.COLOR_HSV2BGR)