Update script.py

script.py

@@ -8,13 +8,12 @@ TEST_IMAGE_PATH = "/tmp/data/test_images"
 SUBMISSION_SAVE_PATH = "submission.csv"
 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.05
+# ✅ 1. Balanced Confidence
+# 0.10 captures weaker tools but keeps precision reasonable.
+CONF_THRESHOLD = 0.10
 
 def get_category_id(cls_id):
-    # IDs 0, 1, 2 are correct based on your experiments.
+    # IDs 0, 1, 2 are correct.
     return int(cls_id)
 
 def run_inference(model, image_path, conf_threshold, save_path):
@@ -37,14 +36,14 @@ def run_inference(model, image_path, conf_threshold, save_path):
         for image_name in test_images:
             full_path = os.path.join(image_path, image_name)
             
-            # 🚀 OPTIMIZED INFERENCE SETTINGS
+            # 🚀 BALANCED TTA SETTINGS
             results = model.predict(
                 full_path, 
                 conf=conf_threshold, 
                 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
+                augment=True,     # ⬅️ ENABLED: Test Time Augmentation (Boosts accuracy)
+                iou=0.60,         # Merge overlapping boxes
+                max_det=30,       # ⬅️ INCREASED: 5 was too strict, 30 is safe
                 verbose=False
             )