python读取二进制数据(包括图片)

1.检索时遇到的问题

# 原方法
def frombytes(mode, size, data, decoder_name="raw", *args):
	pass
# 示例代码
import requests
from PIL import Image
content = requests.get("http://my.cnki.net/Register/CheckCode.aspx?id=1563416120154").content
image = Image.frombytes(mode="RGBA",size=(64,25),data=content,decoder_name="raw")

该代码会抛出异常

ValueError: not enough image data

经查找资料(https://stackoverflow.com/questions/8328198/pil-valueerror-not-enough-image-data) 得知该图片为jpg格式,包括了图片的原始(jpg压缩后的)数据和(jpg)文件头,而frombytes只能读取纯二进制数据

2.解决方案(首先检索到的方式)

from io import StringIO,BytesIO
import requests
from PIL import Image
content = requests.get("http://my.cnki.net/Register/CheckCode.aspx?id=1563416120154").content
image = Image.open(BytesIO(content))

将Image对象转成bytes或者BytesIO

 imgByteArr = BytesIO()
 image.save(imgByteArr,format="PNG")
 imgByteArr.getvalue()

3.图片和二进制转换的三种方式(找到搜索方向后检索到更全的解决方法)

3.1PIL格式转二进制

先读取为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)

3.2数组转二进制

先用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)

3.3图片转二进制

直接读取图片为二进制

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)

参考

Pillow 之frombytes从二进制中读取图片___IProgrammer的博客-CSDN博客

(422条消息) python 图片和二进制转换的三种方式_脸不大的CVer的博客-CSDN博客_python 二进制转图片

你可能感兴趣的:(Python,pillow,python,开发语言)