import numpy as np
import os
import json
import base64
import numpy as np
import six
import cv2
from PIL import Image
from io import BytesIO

#将base64保存为PIL格式的图片
def base64_pil(base_str_path):
    filea = open(base_str_path,'rb')#一定要使用读二进制流的读法. 二进制流在解析的时候不执行编码、解码或换行转换。
    lines = filea.readlines()[0].strip()#去除向txt文件中写入base64文件流时在后面自动添加的空格.
    img=base64.b64decode(lines)#此处解析出来的是二进制文件流.
    image=BytesIO(img)
    image=Image.open(image)#Image.open(）得到的img数据类型呢是Image对象，不是普通的数组。cv2.imread()得到的img数据类型是np.array()类型。
    return image

#将base64保存为numpy格式的图片.
def base64_toimg(base_str_path):
    filea = open(base_str_path,'rb')#一定要使用读二进制流的读法. 
    lines = filea.readlines()[0].strip()#去除向txt文件中写入base64文件流时在后面自动添加的空格.
    img=base64.b64decode(lines)#此处解析出来的是二进制文件流.
    file=open('0611.jpg','wb')#以二进制流的方式保存图片.
    file.write(img)
    file.close()

img=base64_pil('0608_test_img/cc.txt')#此处的img是Image对象.不是普通的数组.
bgr_img=cv2.cvtColor(np.asarray(img),cv2.COLOR_RGB2BGR)
cv2.imwrite('0613.jpg',bgr_img)


#将图片转化为base64格式.
def image_tobase64(img_path):
    pic = open(img_path, "rb")
    pic_base64 = base64.b64encode(pic.read())

    print(pic_base64)

image_tobase64('0613.jpg')