python - 解码 Base64 时出现黑色图像

Question

当我解码 Base64 编码的裁剪面部图像时，它变黑并且数据损坏。

这是一段代码:

def create_learn_msg(db_name, person_name, last_location, images, labels):
    json_string = {}
    cv2.imwrite('%s/%s.png' % ("faces/", "original-image"), images[1])
    print(type(images[1]))  # <type 'numpy.ndarray'>
    image_string = images[1].tostring()
    encode_string = base64.b64encode(image_string)
    # Decode test
    decode_string = base64.b64decode(encode_string)
    decode_image = numpy.fromstring(decode_string, dtype=numpy.uint8)

    print(type(decode_image))  # <type 'numpy.ndarray'>
    cv2.imwrite('%s/%s.png' % ("faces/", "decode-image"), decode_image)

    json_string['data'] = {'type': "learn", 'db_name': db_name, 'person_name': person_name,
                           'last_location': last_location, 'image1': base64.b64encode(images[0]),
                           'image2': base64.b64encode(images[1]), 'label1': labels[0], 'label2': labels[1]}

    return json.dumps(json_string)

Answer 1

我找到了解决方案，我们使用的是 OpenCV 图像，因此我们应该在 Base64 编码之前添加 cv2.imencode ，在 Base64 解码之后添加 cv2.imdecode ，如下所示:

retval, image_string = cv2.imencode('.png', images[1])
encode_string = base64.b64encode(image_string)
decode_string = base64.b64decode(encode_string)
decode_image = numpy.fromstring(decode_string, dtype=numpy.uint8)
original_image = cv2.imdecode(decode_image, 1)