ep 2 v2.8
Browse files
app.py
CHANGED
|
@@ -98,7 +98,13 @@ async def pic_to_uml(
|
|
| 98 |
raise HTTPException(503, "YOLO модель не загружена")
|
| 99 |
|
| 100 |
# В эндпоинте (app.py) измени вызов YOLO:
|
| 101 |
-
results = yolo_model(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 102 |
|
| 103 |
ocr_text = run_ocr(pil_img)
|
| 104 |
|
|
|
|
| 98 |
raise HTTPException(503, "YOLO модель не загружена")
|
| 99 |
|
| 100 |
# В эндпоинте (app.py) измени вызов YOLO:
|
| 101 |
+
results = yolo_model(
|
| 102 |
+
img_cv,
|
| 103 |
+
conf=0.15, # сильно ниже — 0.15–0.20 часто спасает
|
| 104 |
+
iou=0.4, # чуть ниже, чтобы не подавлять близкие боксы
|
| 105 |
+
max_det=300, # по умолчанию 300, но можно 1000 (если много элементов)
|
| 106 |
+
imgsz=1024
|
| 107 |
+
)
|
| 108 |
|
| 109 |
ocr_text = run_ocr(pil_img)
|
| 110 |
|