二进制流转图片

1. 二进制流转图片 （opencv.imdecode）

# -- coding:utf-8 
import cv2
import numpy as np
from matplotlib import pyplot as plt
from PIL import Image
img_url = r'/home/lc/文档/from_thinkpad/windows/source_code/media/3.jpeg'
with open(img_url, 'rb') as f:
    a=f.read()
    
# 二进制数据流转np.ndarray [np.uint8: 8位像素]
img = cv2.imdecode(np.frombuffer(a, np.uint8), cv2.IMREAD_COLOR)
# # 将bgr转为rbg
rgb_img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
print(rgb_img)
# np.ndarray转IMAGE
a = Image.fromarray(rgb_img)
print(a)
# 显示图片
a.show()

# 2、使用io.BytesIO
import io
# 将字节对象转为Byte字节流数据,供Image.open使用
byte_stream = io.BytesIO(a)  
print(type(byte_stream))
roiImg = Image.open(byte_stream)  
# 图片保存 
roiImg.save(r'test.png')
 
# 小结：cv2和io.BytesIO相比，多了一步bgr转rbg，可能使用io.BytesIO更加方便。

 

与直接二进制文件write成的jpg文件相比，再次通过cv2.imread()读入时是相等的

 

ndarray 图片格式 tobytes与bytes之间的转换

import cv2
import matplotlib.pyplot as plt
import numpy as np
import math

img_cv2 = cv2.imread(img_path)
shape = img_cv2.shape
plt.imshow(img_cv2)
plt.show()

img_cv2_bytes = img_cv2.tobytes()

ndarray = np.frombuffer(img_cv2_bytes, dtype=np.uint8)
img_from_bytes=ndarray.reshape(shape)
plt.imshow(img_from_bytes)
plt.show()