如何在fastAPI中返回图像？

  • Z时代
  • 2024-01-10
  • 分类：问答

使用python模块fastAPI，我不知道如何返回图像。在烧瓶中，我将执行以下操作：

@app.route("/vector_image", methods=["POST"])
def image_endpoint():
    # img = ... # Create the image here
    return Response(img, mimetype="image/png")

这个模块中对应的调用是什么？

回答：

我有一个类似的问题，但带有cv2图像。这可能对其他人有用。使用StreamingResponse。

import io
from starlette.responses import StreamingResponse
app = FastAPI()
@app.post("/vector_image")
def image_endpoint(*, vector):
    # Returns a cv2 image array from the document vector
    cv2img = my_function(vector)
    res, im_png = cv2.imencode(".png", cv2img)
    return StreamingResponse(io.BytesIO(im_png.tobytes()), media_type="image/png")

以上是 如何在fastAPI中返回图像？ 的全部内容， 来源链接： utcz.com/qa/405348.html

回到顶部