将numpy.ndarray转换为OpenCV图像/新建图像

需求

将类型为numpy.ndarray的r,g,b值转换成OpenCV图像

环境

python: 3.6.3
opencv: 3.4.1 (已安装opencv-contrib)

关键点

numpy.ndarray中的每个元素的dtype应该为numpy.uint8

验证过程


首先验证一个事情,OpenCV的基础数据类型为numpy.ndarray。


正确代码如下:

import cv2
import numpy as np

# 图片的分辨率为200*300,这里b, g, r设为随机值,注意dtype属性
b = np.random.randint(0, 255, (200, 300), dtype=np.uint8)
g = np.random.randint(0, 255, (200, 300), dtype=np.uint8)
r = np.random.randint(0, 255, (200, 300), dtype=np.uint8)

# 合并通道,形成图片
img = cv2.merge([b, g, r])

# 显示图片
cv2.imshow(test, img)
cv2.waitKey(0)
cv2.destroyWindow(test)

最终结果如下:

但是,当不指定numpy.ndarray``中每个元素的dtype`属性时,显示出来的图片是下面这样的:

结论

结论是显而易见的,将numpy.ndarray转换为OpenCV图像的关键点就是numpy.ndarray中的每个元素的dtype应该为numpy.uint8。当然这样做也相当于新建了图像哈。简单方便。

经验分享 程序员 微信小程序 职场和发展