Python OpenCV从字节字符串加载图像

Question

我正在尝试像PHP函数一样从字符串加载图像 imagecreatefromstring

我怎样才能做到这一点？

我有MySQL Blob字段图像。我正在使用 MySQLdb， 并且不想在PyOpenCV中创建用于处理图像的临时文件。

注意：需要cv（不是cv2）包装函数

Answer 1

这就是我通常用来将数据库中存储的图像转换为Python中的OpenCV图像的方式。

import numpy as np
import cv2
from cv2 import cv

# Load image as string from file/database
fd = open('foo.jpg')
img_str = fd.read()
fd.close()

# CV2
nparr = np.fromstring(img_str, np.uint8)
img_np = cv2.imdecode(nparr, cv2.CV_LOAD_IMAGE_COLOR) # cv2.IMREAD_COLOR in OpenCV 3.1

# CV
img_ipl = cv.CreateImageHeader((img_np.shape[1], img_np.shape[0]), cv.IPL_DEPTH_8U, 3)
cv.SetData(img_ipl, img_np.tostring(), img_np.dtype.itemsize * 3 * img_np.shape[1])

# check types
print type(img_str)
print type(img_np)
print type(img_ipl)

我已添加了从numpy.ndarray到的转换cv2.cv.iplimage，因此上面的脚本将显示：

<type 'str'>
<type 'numpy.ndarray'>
<type 'cv2.cv.iplimage'>

编辑： 从最新的numpy开始1.18.5 +，np.fromstring发出警告，因此np.frombuffer应在该位置使用。