python - return numpy array as image from fastapi

Question

I load an image with img = imageio.imread('hello.jpg'). I want to return this numpy array as image. I know I can do return FileResponse('hello.jpg'). But in the future I will have the pictures as numpy arrays.

How can I return the numpy array img from fastapi in a way that it is equivalent to return FileResponse('hello.jpg')?

question from:https://stackoverflow.com/questions/65903231/return-numpy-array-as-image-from-fastapi

Answer 1

You can use StreamingResponse (https://fastapi.tiangolo.com/advanced/custom-response/#using-streamingresponse-with-file-like-objects) to do it e.g., but before you will need to convert your numpy array to the io.BytesIO or io.StringIO