How to convert the uploaded image to Numpy array?

Example Code To Get an Image

from fastapi import FastAPI, UploadFile, File, Form

app = FastAPI()


@app.post("/")
def read_root(file: bytes = File(...)):
    return {"Hello": "World"}

Or

from fastapi import FastAPI, UploadFile, File, Form

app = FastAPI()


@app.post("/")
def read_root(file: UploadFile = File(...)):
    return {"Hello": "World"}

Environment

• OS: Windows 10
• FastAPI Version == 0.61.2
• Python version: 3.7

Question

How to convert the uploaded image to Numpy array? So it can be then used in libraries like openCV, tensorflow for Computer Vision or Deep Learning Applications.

Things I have already tried

from fastapi import FastAPI, UploadFile, File, Form
from PIL import Image
from io import BytesIO
import numpy as np

app = FastAPI()

def read_imagefile(data) -> Image.Image:
    image = Image.open(BytesIO(data))
    return image

@app.post("/")
async def read_root(file: UploadFile = File(...)):
    image = read_imagefile(await file.read())
    return {"Hello": "World"}

And

from fastapi import FastAPI, UploadFile, File, Form
from PIL import Image
from io import BytesIO
import numpy as np

app = FastAPI()

def load_image_into_numpy_array(data):
    return np.array(Image.open(BytesIO(data)))


@app.post("/")
async def read_root(file: UploadFile = File(...)):
    image = load_image_into_numpy_array(await file.read())
    return {"Hello": "World"}

But I keep getting errors. And none of the solutions in any other thread works. Errors: AttributeError: 'JpegImageFile' object has no attribute 'read' or AttributeError: 'bytes' object has no attribute 'read' or AttributeError: 'numpy.ndarray' object has no attribute 'read' or UnicodeDecodeError: 'utf-8' codec can't decode byte 0xff in position 0: invalid start byte

[tiangolo]

Thanks for the help here @includeamin ! 👏 🙇

Thanks for reporting back and closing the issue @pavitrashah 👍

[maazbin]

Thanks for the code. It worked and solved my problem.