Raise configured YOLO max detections

2026-07-09 10:14:07 +02:00
parent 9bef905da8
commit 25f098bde0
16 changed files with 82 additions and 1 deletions
@@ -31,6 +31,7 @@ class Settings(BaseSettings):
    yolo_device: str = Field(default="cpu", validation_alias="YOLO_DEVICE")
    yolo_image_size: int = Field(default=640, validation_alias="YOLO_IMAGE_SIZE")
    yolo_max_tiles: int = Field(default=100, validation_alias="YOLO_MAX_TILES")
+    yolo_max_detections: int = Field(default=1000, validation_alias="YOLO_MAX_DETECTIONS")
    yolo_batch_size: int = Field(default=1, validation_alias="YOLO_BATCH_SIZE")
    cors_origins: list[str] | str = Field(
        default=["http://localhost:5173", "http://127.0.0.1:5173"],
@@ -72,6 +72,7 @@ class YoloDetectionAdapter:
                    conf=float(confidence_threshold),
                    imgsz=int(self.settings.yolo_image_size),
                    device=self.settings.yolo_device,
+                    max_det=int(self.settings.yolo_max_detections),
                    verbose=False,
                )
        except AppError: