Python - 使用 OpenCV 将字节图像转换为 NumPy 数组

Posted

技术标签:

【中文标题】Python - 使用 OpenCV 将字节图像转换为 NumPy 数组【英文标题】:Python - byte image to NumPy array using OpenCV 【发布时间】:2018-09-05 19:05:48 【问题描述】:

我有一个以字节为单位的图像:

print(image_bytes)

b'\xff\xd8\xff\xfe\x00\x10Lavc57.64.101\x00\xff\xdb\x00C\x00\x08\x04\x04\x04\x04\x04\x05\x05\x05\x05\x05\x05\x06\x06\x06\x06\x06\x06\x06\x06\x06\x06\x06\x06\x06\x07\x07\x07\x08\x08\x08\x07\x07\x07\x06\x06\x07\x07\x08\x08\x08\x08\t\t\t\x08\x08\x08\x08\t\t\n\n\n\x0c\x0c\x0b\x0b\x0e\x0e\x0e\x11\x11\x14\xff\xc4\x01\xa2\x00\x00\x01\x05\x01\x01\x01\x01\x01\x01\x00\x00\x00\x00\x00\x00\x00\x00\x01\x02\x03\x04\x05\x06\x07\x08\t\n\x0b\x01\x00\x03\x01\x01\x01\x01\x01\x01\x01\x01\x01\x00\x00\ ... some other stuff

我可以使用 Pillow 将其转换为 NumPy 数组:

image = numpy.array(Image.open(io.BytesIO(image_bytes))) 

但我不太喜欢使用 Pillow。有没有办法使用清晰的OpenCV,或者直接使用更好的NumPy,或者其他更快的库?

【问题讨论】:

Convert a byte arry to OpenCV image in C++的可能重复 感谢您的反馈,虽然那是 c++,而不是 python。 是的,但它与 python 的代码相同,只是括号更少...docs.opencv.org/3.0-beta/modules/imgcodecs/doc/… @Norrius 我读到“我有以字节为单位的图像”和“image_bytes”,并且不得不做出假设。如果不是这种情况,那么 OP 需要在问题中澄清它为什么不同。 这是一个 JPEG 图像,因为ff,d8,ff 是 JPEG 签名,所以你需要imdecode() 它。看这里...***.com/a/49492989/2836621 【参考方案1】:

我创建了一个2x2 JPEG image 来测试它。图像具有白色、红色、绿色和紫色像素。我用cv2.imdecodenumpy.frombuffer

import cv2
import numpy as np

f = open('image.jpg', 'rb')
image_bytes = f.read()  # b'\xff\xd8\xff\xe0\x00\x10...'

decoded = cv2.imdecode(np.frombuffer(image_bytes, np.uint8), -1)

print('OpenCV:\n', decoded)

# your Pillow code
import io
from PIL import Image
image = np.array(Image.open(io.BytesIO(image_bytes))) 
print('PIL:\n', image)

这似乎可行,尽管通道顺序是 BGR 而不是 PIL.Image 中的 RGB。您可能会使用一些标志来调整它。测试结果:

OpenCV:
 [[[255 254 255]
  [  0   0 254]]

 [[  1 255   0]
  [254   0 255]]]
PIL:
 [[[255 254 255]
  [254   0   0]]

 [[  0 255   1]
  [255   0 254]]]

【讨论】:

在 Keras 中有没有 cv2.imdecode 的替代品?【参考方案2】:

我在网上搜索了最后我解决了:

NumPy 数组(cv2 图像)- 转换

NumPy 到字节

NumPy 的字节数

:.

#data = cv2 image array
def encodeImage(data):
    #resize inserted image
    data= cv2.resize(data, (480,270))
    # run a color convert:
    data= cv2.cvtColor(data, cv2.COLOR_BGR2RGB)
    return bytes(data) #encode Numpay to Bytes string



def decodeImage(data):
    #Gives us 1d array
    decoded = np.fromstring(data, dtype=np.uint8)
    #We have to convert it into (270, 480,3) in order to see as an image
    decoded = decoded.reshape((270, 480,3))
    return decoded;

# Load an color image
image= cv2.imread('messi5.jpg',1)

img_code = encodeImage(image) #Output: b'\xff\xd8\xff\xe0\x00\x10...';
img = decodeImage(img_code) #Output: normal array
cv2.imshow('image_deirvlon',img);
print(decoded.shape)

您可以从here获取完整代码

【讨论】:

以上是关于Python - 使用 OpenCV 将字节图像转换为 NumPy 数组的主要内容,如果未能解决你的问题,请参考以下文章

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

python下使用OpenCV实现计算机视觉读书笔记2图像与字节的变换

python opencv3 图像与原始字节转换

50 行代码,看 Python + OpenCV 玩转实时图像处理!

[转]opencv3 图像处理 之 图像缩放( python与c++实现 )

opencv如何将lena.jpg的灰度值读取出来并在计算机上保存下来