python 图片与base64互相转换

import cv2
import base64
import numpy as np

def img_2_base64(image):
    image = cv2.imencode('.jpg', image)[1]
    image_code = base64.b64encode(image).decode('utf-8')
    return image_code


def base64_2_img(base64_data):
    imgData = base64.b64decode(base64_data)
    nparr = np.frombuffer(imgData, np.uint8)
    img_np = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
    return img_np


if __name__ == "__main__":
    img = cv2.imread(r"test.jpg")
    res = img_2_base64(img)

    with open("1.txt", "w") as f:  # base64 字符串 存入 1.txt
        f.write(res)

    with open("1.txt", "r") as f:  # 打开文件
        data = f.read() 
        img = base64_2_img(data)
        cv2.imwrite("test_res.jpg", img)