PIL格式转二进制

先读取为PIL格式,再转为二进制

import io
import base64
from PIL import Image

def image2byte(image):
    '''
    图片转byte
    image: 必须是PIL格式
    image_bytes: 二进制
    '''
    # 创建一个字节流管道
    img_bytes = io.BytesIO()
    # 将图片数据存入字节流管道, format可以按照具体文件的格式填写
    image.save(img_bytes, format="JPEG")
    # 从字节流管道中获取二进制
    image_bytes = img_bytes.getvalue()
    return image_bytes

def byte2image(byte_data):
    '''
    byte转为图片
    byte_data: 二进制
    '''
    image = Image.open(io.BytesIO(byte_data))
    return image

调用代码

image_path = "img/3.jpg"
image = Image.open(image_path)
byte_data = image2byte(image)
image2 = byte2image(byte_data)

数组转二进制

先用opencv读取为数组格式,再转为二进制

def numpy2byte(image):
    '''
    数组转二进制
    image : numpy矩阵/cv格式图片
    byte_data:二进制数据
    '''
    #对数组的图片格式进行编码
    success,encoded_image = cv2.imencode(".jpg",image)
    #将数组转为bytes
    byte_data = encoded_image.tobytes()
    return byte_data
def byte2numpy(byte_data):
    '''
    byte转numpy矩阵/cv格式
    byte_data:二进制数据
    image : numpy矩阵/cv格式图片
    '''
    image = np.asarray(bytearray(byte_data), dtype="uint8")
    image = cv2.imdecode(image, cv2.IMREAD_COLOR)
    return image

调用代码

image_path = "img/3.jpg"
image = cv2.imread(image_path)
byte_data = numpy2byte(image)
image2 = byte2numpy(byte_data)

图片转二进制

直接读取图片为二进制

import cv2
import numpy as np

def read2byte(path):
    '''
    图片转二进制
    path:图片路径
    byte_data:二进制数据
    '''
    with open(path,"rb") as f:
        byte_data = f.read()
    return byte_data

def byte2numpy(byte_data):
    '''
    byte转numpy矩阵/cv格式
    byte_data:二进制数据
    image : numpy矩阵/cv格式图片
    '''
    image = np.asarray(bytearray(byte_data), dtype="uint8")
    image = cv2.imdecode(image, cv2.IMREAD_COLOR)
    return image

调用代码

image_path = "img/3.jpg"
byte_data = read2byte(image_path)
image = byte2numpy(byte_data)

延伸

有时候数据传输规定转换为base64格式,可以参考我的另一篇文章 图片和base64转换的三种方式

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐