diff --git a/utils/flask_rest_api/restapi.py b/utils/flask_rest_api/restapi.py index 62adb4bbf716..7e7b900107b5 100644 --- a/utils/flask_rest_api/restapi.py +++ b/utils/flask_rest_api/restapi.py @@ -21,12 +21,16 @@ def predict(): return if request.files.get("image"): - image_file = request.files["image"] - image_bytes = image_file.read() + # Method 1 + # with request.files["image"] as f: + # im = Image.open(io.BytesIO(f.read())) - img = Image.open(io.BytesIO(image_bytes)) + # Method 2 + im_file = request.files["image"] + im_bytes = im_file.read() + im = Image.open(io.BytesIO(im_bytes)) - results = model(img, size=640) # reduce size=320 for faster inference + results = model(im, size=640) # reduce size=320 for faster inference return results.pandas().xyxy[0].to_json(orient="records")