使用python flask显示OpenCV图像

Question

我正在使用opencv和python flask API对图像进行一些处理。我想在浏览器中显示图像。

import cv2
from flask import Flask, request, make_response
import base64
import numpy as np
import urllib

app = Flask(__name__)


@app.route('/endpoint', methods=['GET'])
def process():
    image_url = request.args.get('imageurl')
    requested_url = urllib.urlopen(image_url)
    image_array = np.asarray(bytearray(requested_url.read()), dtype=np.uint8)
    img = cv2.imdecode(image_array, -1)


    # Do some processing, get output_img

    retval, buffer = cv2.imencode('.png', output_img)
    png_as_text = base64.b64encode(buffer)
    response = make_response(png_as_text)
    response.headers['Content-Type'] = 'image/png'
    return response

if __name__ == '__main__':
    app.run(debug=True)

但是，我得到的是空的无效图像作为输出。我究竟做错了什么？

Answer 1

如注释中所述，您需要返回图像的字节，而不是base64字符串。

请尝试以下操作：

retval, buffer = cv2.imencode('.png', output_img)
response = make_response(buffer.tobytes())