如何在 fastapi 的一次响应中 return 图像和 json？

Question

我得到一张图片，对其进行更改，然后使用神经网络对其进行分类，应该 return 一张新图片和 json 一个响应。如何用一个端点做到这一点？图像是 return 使用流式响应编辑的，但如何向其添加 json？

import io
from starlette.responses import StreamingResponse

app = FastAPI()

@app.post("/predict")
def predict(file: UploadFile = File(...)):
    img = file.read()
    new_image = prepare_image(img)
    result = predict(new_image)
    return StreamingResponse(io.BytesIO(new_image.tobytes()), media_type="image/png")

Answer 1

我在回复 headers 中添加了 json。更改自：

@app.post("/predict")
def predict(file: UploadFile = File(...)):
    img = file.read()
    new_image = prepare_image(img)
    result = predict(new_image)
    return StreamingResponse(io.BytesIO(new_image.tobytes()), media_type="image/png")

至

@app.post("/predict/")
def predict(file: UploadFile = File(...)):
    file_bytes = file.file.read()
    image = Image.open(io.BytesIO(file_bytes))
    new_image = prepare_image(image)
    result = predict(image)
    bytes_image = io.BytesIO()
    new_image.save(bytes_image, format='PNG')
    return Response(content = bytes_image.getvalue(), headers = result, media_type="image/png")

如何在 fastapi 的一次响应中 return 图像和 json？

How to return image and json in one response in fastapi?

python

fastapi