diff --git a/4_detect.py b/4_detect.py index 79e824d..f46e4f2 100644 --- a/4_detect.py +++ b/4_detect.py @@ -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")