@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 上返回它,但策略是什么?
尝试使用简单的 pred.tolist() 返回 fastapi 可以理解的普通列表。