handsme commited on
Commit ·
689f815
1
Parent(s): 9595b9e
修复:API 端点添加图片缩放防止大图崩溃
Browse files
app.py
CHANGED
|
@@ -89,6 +89,13 @@ async def api_predict(request: Request):
|
|
| 89 |
image_bytes = base64.b64decode(base64_str)
|
| 90 |
image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
|
| 91 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 92 |
result_json_str = ocr_recognize(image)
|
| 93 |
return {"data": [result_json_str]}
|
| 94 |
except Exception as e:
|
|
|
|
| 89 |
image_bytes = base64.b64decode(base64_str)
|
| 90 |
image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
|
| 91 |
|
| 92 |
+
# 限制图片最大边为 2000px,防止大图导致 PaddlePaddle C++ 层内存溢出
|
| 93 |
+
max_side = 2000
|
| 94 |
+
w, h = image.size
|
| 95 |
+
if max(w, h) > max_side:
|
| 96 |
+
scale = max_side / max(w, h)
|
| 97 |
+
image = image.resize((int(w * scale), int(h * scale)), Image.LANCZOS)
|
| 98 |
+
|
| 99 |
result_json_str = ocr_recognize(image)
|
| 100 |
return {"data": [result_json_str]}
|
| 101 |
except Exception as e:
|