【flask】——接收和返回图片

转载自:https://www.jianshu.com/p/cafbeec49674

接收图片

首先写一个接收图片的接口，

@img_tansfrom.route('/img/send_img', methods=['POST'])
def send_img():
    f = request.files['file']
    img = f.read()
    print('img',type(img))
    im1 = Image.open(img)
    print('im1',type(im1))
    im2 = cv2.imread(img)
    print('im2',type(im2))
    return 'Send Img Test'

对于接收到的图片，显示的类型为

img <class 'bytes'>

接收到图片的已经是2进制的文件，这里对于图像的读取没有问题。
但是下面用两个常用图像库打开的时候分别报错：

Image.open()
ValueError: embedded null byte
# cv2.imread()
SystemError: <built-in function imread> returned NULL without setting an error
对应的解决办法也很简单：
使用cv2读取图片

im1 = cv2.imdecode(np.frombuffer(img, np.uint8), cv2.IMREAD_COLOR)

# np.ndarray转IMAGE
# im = Image.fromarray(img_cv2)
使用Image读取图片

import io
byte_stream = io.BytesIO(img)  
im2 = Image.open(byte_stream) 
修改以后，重启服务，发送图片测试，结果如下：

img <class 'bytes'>
im1 <class 'PIL.PngImagePlugin.PngImageFile'>
im2 <class 'numpy.ndarray'>

返回图片

接收到了传来的图片下一步就是处理完图片过后传回去，一种方式比较简单，就是把图像转成二进制的形式传回去，用户端接收的也是一堆二进制数字，还需要后续的转换才能看图片。那如果想直接从浏览器看到返回的图片，需要使用flask的send_file方法。

return send_file(
        io.BytesIO(imgByteArr),
        mimetype='image/png',
        as_attachment=True,
        attachment_filename='result.jpg'
    )

这里的imgByteArr是使用Image读取处理后的图片

def draw_img(img):
    '''对读取的图片进行处理'''
    img_stream = io.BytesIO(img)
    img = Image.open(img_stream)
    # 图像处理逻辑
    #
    #

    imgByteArr = io.BytesIO()
    img.save(imgByteArr,format='PNG')
    imgByteArr = imgByteArr.getvalue()

    return imgByteArr

这样启动服务后，服务器接收到请求，可以直接在浏览器返回图片。