import io
import base64
from PIL import Image

def image2byte(image):
    '''
    圖片轉byte
    image: 必須是PIL格式
    image_bytes: 二進制
    '''
    # 創(chuàng)建一個字節(jié)流管道
    img_bytes = io.BytesIO()
    # 將圖片數(shù)據(jù)存入字節(jié)流管道， format可以按照具體文件的格式填寫
    image.save(img_bytes, format="JPEG")
    # 從字節(jié)流管道中獲取二進制
    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)

def numpy2byte(image):
    '''
    數(shù)組轉二進制
    image : numpy矩陣/cv格式圖片
    byte_data：二進制數(shù)據(jù)
    '''
    #對數(shù)組的圖片格式進行編碼
    success,encoded_image = cv2.imencode(".jpg",image)
    #將數(shù)組轉為bytes
    byte_data = encoded_image.tobytes()
    return byte_data
def byte2numpy(byte_data):
    '''
    byte轉numpy矩陣/cv格式
    byte_data：二進制數(shù)據(jù)
    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：二進制數(shù)據(jù)
    '''
    with open(path,"rb") as f:
        byte_data = f.read()
    return byte_data

def byte2numpy(byte_data):
    '''
    byte轉numpy矩陣/cv格式
    byte_data：二進制數(shù)據(jù)
    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)