Update script.py

script.py

@@ -11,7 +11,7 @@ MODEL_WEIGHTS = "best.pt"
 # ✅ 1. Sane Confidence: 
 # 0.20 was safe. We go slightly lower to 0.15 to improve Recall, 
 # relying on 'max_det' to fix the Precision.
-CONF_THRESHOLD = 0.15
+CONF_THRESHOLD = 0.05
 
 def get_category_id(cls_id):
     # IDs 0, 1, 2 are correct based on your experiments.
@@ -41,10 +41,10 @@ def run_inference(model, image_path, conf_threshold, save_path):
             results = model.predict(
                 full_path, 
                 conf=conf_threshold, 
-                imgsz=1024,       # Match training size
-                augment=False,    # Disable to prevent blurring/drift
-                iou=0.50,         # Aggressive NMS to merge overlapping boxes
-                max_det=20,       # ⬅️ CRITICAL: Limit to top 20 objects per image
+                imgsz=1024,       # Keep High Res
+                augment=False,    
+                iou=0.60,         # Slightly higher IoU to keep close instruments
+                max_det=5,        # ⬅️ CRITICAL: Only allow top 5 instruments per image
                 verbose=False
             )