如何在 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 理解的普通列表。