Python读取base64图片数据

Python读取base64图片数据

在Python中,可以使用base64模块来解码Base64编码的字符串,并将其转换回原始的数据(如图片)。但通常,Base64编码的图片字符串会带有前缀(如data:image/jpeg;base64,),需要先去掉这个前缀再进行解码。

以下是一个简单的例子,展示如何从Base64编码的字符串中读取并保存图片:

python 复制代码
import base64
import io
from PIL import Image

# 假设有一个Base64编码的图片字符串,这里我们使用一个简化的示例
base64_str = "data:image/jpeg;base64,/9j/4AAQSkZJRgABAQEAAAAAAAD/4gIoSUNDX1BST0ZJTEUAAQEAAAIYAAAAAAIQAABtbnRyUkdCIFhZJiouMjY7...(此处省略其他Base64编码的内容)"

# 去掉前缀('data:image/jpeg;base64,')
base64_data = base64_str.split(',')[1]

# 解码Base64数据
image_bytes = base64.b64decode(base64_data)

# 将字节数据转换为BytesIO对象
image_io = io.BytesIO(image_bytes)

# 使用PIL库读取图片
image = Image.open(image_io)

# 保存图片到文件
image.save('output.jpg')

注意:上面的Base64字符串base64_str只是一个示例,需要将其替换为实际的Base64编码的图片字符串。而且,Base64字符串可能以不同的MIME类型开始(如data:image/png;base64,),需要相应地调整代码以匹配的字符串。

使用场景:

以下是百度Aistudio部署的应用为例:

python 复制代码
# 获取base64图片数据
import requests


def query(payload, token='你自己的token令牌', appCode='你的appCode'):
    API_URL = f"https://{appCode}.aistudio-hub.baidu.com/image/generations"
    headers = {
        # 请前往 https://aistudio.baidu.com/index/accessToken 查看 访问令牌
        "Authorization": f"token {token}",
        "Content-Type": "application/json"
    }
    response = requests.post(API_URL, headers=headers, json=payload)
    return response.json()

def access_image(prompt):
    output = query({
        "prompt": f"{prompt}"
    })
    return output['data'][0]["b64_image"]
python 复制代码
# base64数据转图片
import base64  
import io  
from PIL import Image  


def decode_b64_image(base64_data):
    # 解码Base64数据  
    image_bytes = base64.b64decode(base64_data)  

    # 将字节数据转换为BytesIO对象  
    image_io = io.BytesIO(image_bytes)  
    
    # 使用PIL库读取图片  
    image = Image.open(image_io)  
    
    # 保存图片到文件  
    # image.save('output.jpg')
    return image
python 复制代码
# 调用函数
prompt = "杰作,高品质,超精细,全细节,8k"
base64_image = access_image(prompt)
decode_b64_image(base64_image)
相关推荐
费弗里2 小时前
Python全栈应用开发利器Dash 3.x新版本介绍(1)
python·dash
李少兄9 天前
解决OSS存储桶未创建导致的XML错误
xml·开发语言·python
就叫飞六吧9 天前
基于keepalived、vip实现高可用nginx (centos)
python·nginx·centos
Vertira9 天前
PyTorch中的permute, transpose, view, reshape和flatten函数详解(已解决)
人工智能·pytorch·python
学Linux的语莫9 天前
python基础语法
开发语言·python
匿名的魔术师9 天前
实验问题记录:PyTorch Tensor 也会出现 a = b 赋值后,修改 a 会影响 b 的情况
人工智能·pytorch·python
Ven%9 天前
PyTorch 张量(Tensors)全面指南:从基础到实战
人工智能·pytorch·python
mahuifa9 天前
PySide环境配置及工具使用
python·qt·环境配置·开发经验·pyside
大熊猫侯佩9 天前
ruby、Python 以及 Swift 语言关于 “Finally” 实现的趣谈
python·ruby·swift
19899 天前
【Dify精讲】第19章:开源贡献指南
运维·人工智能·python·架构·flask·开源·devops