如何在 FastAPI 中返回 NumPy 数组？ [重复]

Question

@app.post("/predict/")
async def predict(file:UploadFile = File(...)):
    bytes_str = io.BytesIO(await file.read())
    img = Image.open(bytes_str)
    img_array = np.array(img.convert('RGB'))
    img_array = cv2.resize(img_array, (1024,1024))
    img_final = np.expand_dims(img_array, axis = 0)
    pred = model.predict(img_final)

此代码返回一个 NumPy 数组，我需要在端点上的 FastAPI 上返回它，但策略是什么？

Answer 1

尝试使用简单的 pred.tolist() 返回 fastapi 可以理解的普通列表。