How to return an image in FastAPI after processing it with OpenCV? [duplicate]

Question

I am trying to return an image in FastAPI after comparing two images using Opencv.

Here's what I have done so far:

from fastapi import FastAPI , File, UploadFile
import numpy as np
from cv2 import *
import os
import base64


app = FastAPI(debug = True)

    
@app.post("/uploadfile/")
async def create_upload_file(file: UploadFile = File(...),file1: UploadFile = File(...)):
    content = await file.read()
    nparr = np.fromstring(content, np.uint8)
    img = cv2.imdecode(nparr, cv2.IMREAD_COLOR)

    content1 = await file1.read()
    nparr1 = np.fromstring(content1, np.uint8)
    img1 = cv2.imdecode(nparr1, cv2.IMREAD_COLOR)

    akaze = cv2.AKAZE_create()
    kpts1, desc1 = akaze.detectAndCompute(img, None)
    kpts2, desc2 = akaze.detectAndCompute(img1, None)
    matcher = cv2.DescriptorMatcher_create(cv2.DescriptorMatcher_BRUTEFORCE_HAMMING)
    matches_1 = matcher.knnMatch(desc1, desc2, 2)
    good_points = []
    for m,n in matches_1:
            if m.distance < 0.7 * n.distance:
                good_points.append(m)
    mat = (round(len(kpts2)/len(good_points),2))

#where I am getting an error

    return_img = cv2.processImage(img)
    _, encoded_img = cv2.imencode('.PNG', return_img)
    encoded_img = base64.b64encode(return_img)

    return {"The similarity is": mat,'encoded_img': endcoded_img}

What am I doing wrong?

@MarceloTrylesinski , I want to return an image using fastapi , is that possible? — SUBHRA SANKHA
– SUBHRA SANKHA, Commented Jun 13, 2020 at 21:42
Does How do I return an image in fastAPI? answer your question? — dbc
– dbc, Commented Jun 14, 2020 at 18:05
Does this answer your question? How do I return an image in fastAPI? — Abu Shoeb
– Abu Shoeb, Commented Jun 29, 2021 at 4:21

Yagiz Degirmenci · Accepted Answer · 2020-11-10 21:30:31Z

26

Yes you can return an image with FastAPI, it's actually so simple.

from fastapi import FastAPI
from fastapi.responses import FileResponse

some_file_path = "some_image.jpeg"
app = FastAPI()


@app.get("/")
async def main():
    return FileResponse(some_file_path)

Make sure to install aiofiles with pip install aiofiles otherwise, you'll be getting an error like this:

AssertionError: 'aiofiles' must be installed to use FileResponse

If you have an image as bytes consider using StreamingResponse

from io import BytesIO

@app.post("/send_image")
async def send():
    image = BytesIO()
    img =                                        # Do something here to create an image
    img.save(image, format='JPEG', quality=85)   # Save image to BytesIO
    image.seek(0)                                # Return cursor to starting point
    return StreamingResponse(image.read(), media_type="image/jpeg")

edited Nov 10, 2020 at 21:30

answered Jul 16, 2020 at 16:45

Yagiz Degirmenci

21.4k9 gold badges75 silver badges92 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Sandrin Joy Over a year ago

i did install aiofiles, but still getting the same error

Adnan Taufique · Accepted Answer · 2020-11-23 06:51:06Z

2

The solution can be found here. You are converting the Opencv Image before encoding it in the 3rd line.

    return_img = cv2.processImage(img)
    _, encoded_img = cv2.imencode('.PNG', return_img)
    encoded_img = base64.b64encode(return_img)

    return {"The similarity is": mat,'encoded_img': endcoded_img}

Replace return_img with encoded_img and everything should be working as expected.

    return_img = cv2.processImage(img)
    _, encoded_img = cv2.imencode('.PNG', return_img)
    encoded_img = base64.b64encode(encoded_img)

    return {"The similarity is": mat,'encoded_img': endcoded_img}

answered Nov 23, 2020 at 6:51

Adnan Taufique

3891 silver badge9 bronze badges

Comments

lsabi · Accepted Answer · 2020-06-14 08:22:30Z

1

If you want to return the image only, then return the image itself with the correct encoding in the header see Return a Response Directly

If you need to also return other information with the image in the json, then see this SO question How do you put an image file in a json object? (note: the solutions are in javascript, but it's quite easy to adapt it for python)

answered Jun 14, 2020 at 8:22

lsabi

4,6161 gold badge20 silver badges30 bronze badges

Collectives™ on Stack Overflow

How to return an image in FastAPI after processing it with OpenCV? [duplicate]

3 Answers 3

1 Comment

Comments

Comments

Linked

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

1 Comment

Comments

Comments

Linked

Related