hr-ocr-bankcard/s_utils/image.py

from base64 import b64decode
import numpy as np
import cv2

def base64_to_np(img_data):
    color_image_flag = 1
    img_data = img_data.split(',',1)[-1]
    img = cv2.imdecode(np.fromstring(b64decode(img_data), dtype=np.uint8), color_image_flag)
    # img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
    return img