import cv2
import base64
import numpy as np

def img_to_base64(img_array):
    # 傳入圖片為RGB格式numpy矩陣，傳出的base64也是通過(guò)RGB的編碼
    img_array = cv2.cvtColor(img_array, cv2.COLOR_RGB2BGR) #RGB2BGR，用于cv2編碼
    encode_image = cv2.imencode(".jpg", img_array)[1] #用cv2壓縮/編碼，轉(zhuǎn)為一維數(shù)組
    byte_data = encode_image.tobytes() #轉(zhuǎn)換為二進(jìn)制
    base64_str = base64.b64encode(byte_data).decode("ascii") #轉(zhuǎn)換為base64
    return base64_str
    
def base64_to_img(base64_str):
    # 傳入為RGB格式下的base64，傳出為RGB格式的numpy矩陣
    byte_data = base64.b64decode(base64_str)#將base64轉(zhuǎn)換為二進(jìn)制
    encode_image = np.asarray(bytearray(byte_data), dtype="uint8")# 二進(jìn)制轉(zhuǎn)換為一維數(shù)組
    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)

from io import BytesIO
from PIL import Image
import base64

def image_to_base64(image):
    # 輸入為PIL讀取的圖片，輸出為base64格式
    byte_data = BytesIO()# 創(chuàng)建一個(gè)字節(jié)流管道
    image.save(byte_data, format="JPEG")# 將圖片數(shù)據(jù)存入字節(jié)流管道
    byte_data = byte_data.getvalue()# 從字節(jié)流管道中獲取二進(jìn)制
    base64_str = base64.b64encode(byte_data).decode("ascii")# 二進(jìn)制轉(zhuǎn)base64
    return base64_str

def base64_to_image(base64_str):
    # 輸入為base64格式字符串，輸出為PIL格式圖片
    byte_data = base64.b64decode(base64_str) # base64轉(zhuǎn)二進(jìn)制
    image = Image.open(BytesIO(byte_data)) # 將二進(jìn)制轉(zhuǎn)為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")# 二進(jìn)制轉(zhuǎn)base64
    return base64_str

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