从 BGR 到灰度的 OpenCV 颜色转换错误

Question

我正在尝试使用以下代码将图像从 BGR 转换为灰度格式：

img = cv2.imread('path//to//image//file')
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

这似乎工作正常：我检查了img变量的数据类型，结果是 numpy ndarray 和形状是(100,80,3) 。 但是，如果我给出与cvtColor函数输入维度相同的本机 numpy ndarray 数据类型的图像，则会出现以下错误：

Error: Assertion failed (depth == 0 || depth == 2 || depth == 5) in cv::cvtColor, file D:\Build\OpenCV\opencv-3.4.1\modules\imgproc\src\color.cpp, line 11109
cv2.error: OpenCV(3.4.1) D:\Build\OpenCV\opencv-3.4.1\modules\imgproc\src\color.cpp:11109: error: (-215) depth == 0 || depth == 2 || depth == 5 in function cv::cvtColor

第二种情况的代码是（在此处制作自定义np.ndarray ）：

img = np.full((100,80,3), 12)
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) 

谁能澄清这个错误的原因是什么以及如何纠正它？

Answer 1

这是因为您的 numpy 数组不是由正确的数据类型组成。 默认情况下np.int64 （64 位）类型的数组，但是， cv2.cvtColor()需要 8 位（ np.uint8 ）或 16 位（ np.uint16 ）。 要更正此更改np.full()函数以包含数据类型：

img = np.full((100,80,3), 12, np.uint8)

Answer 2

出现错误是因为cv2.imread返回的 numpy 数组的数据类型是uint8 ，与np.full()返回的 numpy 数组的数据类型不同。 要将数据类型设为 uint8，请添加dtype参数-

img = np.full((100,80,3), 12, dtype = np.uint8)
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

Answer 3

使用初始图像作为源和dtype=np.uint8初始化新的 numpy 数组可能更容易：

import numpy as np    
img = cv2.imread('path//to//image//file')    
img = np.array(img, dtype=np.uint8)
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

Answer 4

想象一下，如果您尝试将其应用为：

data = np.array(list(map(preprocessing,data)))

它不会工作，因为 np.array 创建 int64 并且您试图将 np.uint8 分配给它，您应该做的是添加 dtype 参数如下：

data = np.array(list(map(preprocessing,data)), dtype = np.uint8)