python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > python图片和base64互转的方式

python关于图片和base64互转的三种方式

作者:脸不大的CVer

无论使用cv2、PIL还是直接读取图片的方法进行图片与Base64的转换,核心步骤都涉及到二进制格式的转换,每种方法的基本过程都是:Base64转二进制,然后二进制转图片,或反向操作,这些方法均基于二进制与图片转换的基本原理

通过cv2进行转换

import cv2
import base64
import numpy as np

def img_to_base64(img_array):
    # 传入图片为RGB格式numpy矩阵,传出的base64也是通过RGB的编码
    img_array = cv2.cvtColor(img_array, cv2.COLOR_RGB2BGR) #RGB2BGR,用于cv2编码
    encode_image = cv2.imencode(".jpg", img_array)[1] #用cv2压缩/编码,转为一维数组
    byte_data = encode_image.tobytes() #转换为二进制
    base64_str = base64.b64encode(byte_data).decode("ascii") #转换为base64
    return base64_str
    
def base64_to_img(base64_str):
    # 传入为RGB格式下的base64,传出为RGB格式的numpy矩阵
    byte_data = base64.b64decode(base64_str)#将base64转换为二进制
    encode_image = np.asarray(bytearray(byte_data), dtype="uint8")# 二进制转换为一维数组
    img_array = cv2.imdecode(encode_image, cv2.IMREAD_COLOR)# 用cv2解码为三通道矩阵
    img_array = cv2.cvtColor(img_array, cv2.COLOR_BGR2RGB)# BGR2RGB
    return img_array

方法调用:

img = np.zeros((100,100,3), dtype=np.uint8)
img[:,:,2]=255
base64_img = img_to_base64(img)
img_from_base64 = base64_to_img(base64_img)

结果显示:

通过PIL进行转换

from io import BytesIO
from PIL import Image
import base64

def image_to_base64(image):
    # 输入为PIL读取的图片,输出为base64格式
    byte_data = BytesIO()# 创建一个字节流管道
    image.save(byte_data, format="JPEG")# 将图片数据存入字节流管道
    byte_data = byte_data.getvalue()# 从字节流管道中获取二进制
    base64_str = base64.b64encode(byte_data).decode("ascii")# 二进制转base64
    return base64_str

def base64_to_image(base64_str):
    # 输入为base64格式字符串,输出为PIL格式图片
    byte_data = base64.b64decode(base64_str) # base64转二进制
    image = Image.open(BytesIO(byte_data)) # 将二进制转为PIL格式图片
    return image

方法调用:

img = np.zeros((100,100,3),dtype=np.uint8)
img[:,:,2]=255
image = Image.fromarray(img)
base64_str = image_to_base64(image)
image_from_base64=base64_to_image(base64_str)

结果显示:

直接读取图片方式

import base64

def path2base64(path):
    with open(path,"rb") as f:
        byte_data = f.read()
    base64_str = base64.b64encode(byte_data).decode("ascii")# 二进制转base64
    return base64_str

方法调用:

base64_img = path2base64("1.jpg")
img_from_base64 = base64_to_img(base64_img)

总结

无论使用哪种方式进行图片和base64的互相转换,都需要中间变量——二进制格式。

基本过程都为:base64 <——> 二进制 <——> 图片

其中使用到了我的另一篇文章二进制和图片互相转换的三种方式

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

您可能感兴趣的文章:
阅读全文