python 图片与二进制之间的转换

一:PIL格式图片转成二进制

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

import io
import base64
from PIL import Image

def image2byte(image):
    '''
    图片转byte
    image: 必须是PIL格式
    image_bytes: 二进制
    '''
    # 创建一个字节流管道
    img_bytes = io.BytesIO()
    #把PNG格式转换成的四通道转成RGB的三通道,然后再保存成jpg格式
    image = image.convert("RGB")
    # 将图片数据存入字节流管道, 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)

你可能感兴趣的:(python,python,opencv,计算机视觉)