Update script.py

script.py

@@ -4,51 +4,41 @@ from ultralytics import YOLO
 import glob
 
 # --- CONFIGURATION ---
-# The evaluation server usually puts test images here
 TEST_IMAGE_PATH = "/tmp/data/test_images"
 SUBMISSION_SAVE_PATH = "submission.csv"
 MODEL_WEIGHTS = "best.pt"
 
-# You can tune this. 0.25 - 0.30 is usually a safe balance for mAP.
-CONF_THRESHOLD = 0.001
+# ✅ FIX 1: Set a balanced threshold. 
+# 0.20 removes the noise but keeps the valid weak detections.
+CONF_THRESHOLD = 0.20
 
 def get_category_id(cls_id):
-    # ❌ OLD: return int(cls_id)
-    # ✅ NEW: Add 1 to match competition format (1, 2, 3)
-    return int(cls_id) + 1
+    # ✅ FIX 2: Revert to original IDs. 
+    # The drop in score proved that adding +1 was wrong.
+    return int(cls_id)
 
 def run_inference(model, image_path, conf_threshold, save_path):
     print(f"🚀 Checking for images in {image_path}...")
     
-    # 1. Get all images (support multiple extensions)
     test_images = []
     if os.path.exists(image_path):
-        # Recursive search just in case, though usually they are flat in the folder
         extensions = ['*.png', '*.jpg', '*.jpeg', '*.bmp']
         for ext in extensions:
             test_images.extend(glob.glob(os.path.join(image_path, ext)))
-        
-        # Sort to ensure consistent order if needed
         test_images.sort()
-        # Just keep filenames for the loop to match your specific structure requirements
         test_images = [os.path.basename(x) for x in test_images]
     else:
-        print(f"⚠️ Warning: {image_path} not found. Testing with empty list (or verify path).")
-        # If running locally for debug, you might want to point to a local folder here
-        # test_images = os.listdir(".") 
+        print(f"⚠️ Warning: {image_path} not found.")
 
     print(f"🔍 Found {len(test_images)} images.")
 
-    # Prepare lists for the dataframe
     df_rows = []
 
-    # 2. Run Inference
     if len(test_images) > 0:
         for image_name in test_images:
             full_path = os.path.join(image_path, image_name)
             
-            # Run YOLOv8 inference
-            # verbose=False keeps the logs clean
+            # Run inference
             results = model.predict(full_path, conf=conf_threshold, verbose=False)
             
             bbox_list = []
@@ -56,50 +46,34 @@ def run_inference(model, image_path, conf_threshold, save_path):
             
             for result in results:
                 for box in result.boxes:
-                    # YOLO format is Center-X, Center-Y, Width, Height
+                    # YOLO (Center) -> COCO (Top-Left)
                     x_c, y_c, w, h = box.xywh[0].tolist()
-                    
-                    # Convert to COCO format: Top-Left-X, Top-Left-Y, Width, Height
                     x_min = x_c - (w / 2)
                     y_min = y_c - (h / 2)
                     
-                    # Store as list [x, y, w, h]
                     bbox_list.append([x_min, y_min, w, h])
                     
-                    # Get predicted class ID (0, 1, or 2)
                     cls_id = int(box.cls[0])
                     category_list.append(get_category_id(cls_id))
 
-            # 3. Format Prediction String
-            # The baseline requires the bbox and category_id columns to be Strings of Lists
             df_rows.append({
                 "file_name": image_name,
-                "bbox": str(bbox_list),           # e.g. "[[10.5, 20.1, 50.0, 50.0]]"
-                "category_id": str(category_list) # e.g. "[0]" or "[0, 2]"
+                "bbox": str(bbox_list),
+                "category_id": str(category_list)
             })
 
-    # 4. Create DataFrame and Save
+    # Save
     df_predictions = pd.DataFrame(df_rows, columns=["file_name", "bbox", "category_id"])
-    
-    # Safety check: if no images found, create an empty CSV with headers to prevent crash
     if df_predictions.empty:
-        print("⚠️ No predictions made (no images found?). Creating empty CSV.")
         df_predictions = pd.DataFrame(columns=["file_name", "bbox", "category_id"])
 
-    print(f"💾 Saving {len(df_predictions)} rows to {save_path}...")
     df_predictions.to_csv(save_path, index=False)
     print("✅ Done!")
 
 if __name__ == "__main__":
-    # Ensure ultralytics is installed via requirements.txt before this runs
-    print(f"🔥 Loading YOLOv8 model: {MODEL_WEIGHTS}...")
-    
     try:
-        # Load the trained model
+        print(f"🔥 Loading Model...")
         model = YOLO(MODEL_WEIGHTS)
-        
-        # Execute inference
         run_inference(model, TEST_IMAGE_PATH, CONF_THRESHOLD, SUBMISSION_SAVE_PATH)
-        
     except Exception as e:
-        print(f"❌ Critical Error in main execution: {e}")
+        print(f"❌ Error: {e}")