Flask接收和返回图片

上一篇文章介绍一下基本的flask文件结构，这篇讲一下如何用flask接收和返回图片。包括在接收之后如何用Image或者cv2打开图片做相应处理，已经处理后，如何将图片返回。

接收图片

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

@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 

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

# Image.open()
ValueError: embedded null byte

# cv2.imread()
SystemError:  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 
im1 
im2 

返回图片

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

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

参考资料

Convert PIL Image to byte array?

File Stream - ValueError: embedded null byte