DEFAULT_IMGSZ=800

This commit is contained in:
Hamza Zakaria 2026-05-03 15:19:21 +01:00
parent 69b307cc07
commit 92e4c5dd37

View File

@ -26,7 +26,7 @@ from ultralytics import YOLO
DEFAULT_WEIGHTS = "runs/detect/coffee_v1/weights/best.pt"
DEFAULT_CONF = .35 # lower = more detections and more false positives.
DEFAULT_IOU = .45 # NMS IoU threshold.
DEFAULT_IMGSZ = 640
DEFAULT_IMGSZ = 800
OUTPUT_DIR = Path("detections")