将Numpy数组转换为OpenCV数组
image-processing
numpy
opencv
python
17
0

我正在尝试将代表黑白图像的2D Numpy数组转换为3通道OpenCV数组（即RGB图像）。

基于代码示例和文档，我正在尝试通过Python进行此操作，例如： 

import numpy as np, cv
vis = np.zeros((384, 836), np.uint32)
h,w = vis.shape
vis2 = cv.CreateMat(h, w, cv.CV_32FC3)
cv.CvtColor(vis, vis2, cv.CV_GRAY2BGR)

但是，对CvtColor（）的调用将引发以下cpp级异常： 

OpenCV Error: Image step is wrong () in cvSetData, file /build/buildd/opencv-2.1.0/src/cxcore/cxarray.cpp, line 902
terminate called after throwing an instance of 'cv::Exception'
  what():  /build/buildd/opencv-2.1.0/src/cxcore/cxarray.cpp:902: error: (-13)  in function cvSetData

Aborted

我究竟做错了什么？

参考资料：
Stack Overflow
收藏
评论
共 2 个回答
高赞 时间 活跃

这就是对我有用的... 

import cv2
import numpy as np

#Created an image (really an ndarray) with three channels 
new_image = np.ndarray((3, num_rows, num_cols), dtype=int)

#Did manipulations for my project where my array values went way over 255
#Eventually returned numbers to between 0 and 255

#Converted the datatype to np.uint8
new_image = new_image.astype(np.uint8)

#Separated the channels in my new image
new_image_red, new_image_green, new_image_blue = new_image

#Stacked the channels
new_rgb = np.dstack([new_image_red, new_image_green, new_image_blue])

#Displayed the image
cv2.imshow("WindowNameHere", new_rgbrgb)
cv2.waitKey(0)
收藏
评论

您的代码可以固定如下： 

import numpy as np, cv
vis = np.zeros((384, 836), np.float32)
h,w = vis.shape
vis2 = cv.CreateMat(h, w, cv.CV_32FC3)
vis0 = cv.fromarray(vis)
cv.CvtColor(vis0, vis2, cv.CV_GRAY2BGR)

简短说明：

  1. OpenCV不支持np.uint32数据类型（它支持uint8 ， int8 ， uint16 ， int16 ， int32 ， float32 ， float64 ）
  2. cv.CvtColor无法处理numpy数组，因此必须将两个参数都转换为OpenCV类型。 cv.fromarray进行此转换。
  3. cv.CvtColor两个参数必须具有相同的深度。因此，我将源类型更改为32bit float以匹配destination。

我也建议您使用较新版本的OpenCV python API，因为它使用numpy数组作为主要数据类型： 

import numpy as np, cv2
vis = np.zeros((384, 836), np.float32)
vis2 = cv2.cvtColor(vis, cv2.COLOR_GRAY2BGR)
收藏
评论
新手导航
  • 社区规范
  • 提出问题
  • 进行投票
  • 个人资料
  • 优化问题
  • 回答问题