DEFAULT_IMGSZ=800

This commit is contained in:
Hamza committed 2026-05-03 16:19:21 +02:00
1 parent 69b307cc07
commit 92e4c5dd37
1 file changed
+1 -1
+1 -1
View File
@@ -26,7 +26,7 @@ from ultralytics import YOLO
DEFAULT_WEIGHTS = "runs/detect/coffee_v1/weights/best.pt" DEFAULT_WEIGHTS = "runs/detect/coffee_v1/weights/best.pt"
DEFAULT_CONF = .35 # lower = more detections and more false positives. DEFAULT_CONF = .35 # lower = more detections and more false positives.
DEFAULT_IOU = .45 # NMS IoU threshold. DEFAULT_IOU = .45 # NMS IoU threshold.
DEFAULT_IMGSZ = 640 DEFAULT_IMGSZ = 800
OUTPUT_DIR = Path("detections") OUTPUT_DIR = Path("detections")