import base64
from base64 import b64decode
import numpy as np
import cv2

# base64格式转numpy格式
def base64_to_np(img_data):
    color_image_flag = 1
    img_data = img_data.split(',',1)[-1]
    return cv2.imdecode(np.fromstring(b64decode(img_data), dtype=np.uint8), color_image_flag)

def base64_cv2(base64_str):
    base64_str = base64_str.split(',', 1)[-1]
    img_str = base64.b64decode(base64_str)
    np_arr = np.fromstring(img_str, np.int8)
    image = cv2.imdecode(np_arr, cv2.IMREAD_COLOR)
    return image