gpt4 book ai didi

python - 如何在fastapi的一个响应中返回图像和json?

转载 作者:行者123 更新时间:2023-12-04 08:04:54 30 4
gpt4 key购买 nike

我得到一个图像,更改它,然后使用神经网络对其进行分类,应该返回一个新图像和带有响应的 json。如何用一个端点做到这一点?
图像与流响应一起返回,但如何向其中添加 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")

最佳答案

我在响应头中添加了 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")

关于python - 如何在fastapi的一个响应中返回图像和json?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/66265511/

30 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com