Update script.py

script.py

@@ -39,7 +39,14 @@ def run_inference(model, image_path, conf_threshold, save_path):
             full_path = os.path.join(image_path, image_name)
             
             # Run inference
-            results = model.predict(full_path, conf=conf_threshold, verbose=False)
+            # NEW (Add imgsz=1024)
+            results = model.predict(
+                full_path, 
+                conf=conf_threshold, 
+                imgsz=1024,      # ⬅️ Match your training size
+                augment=True,    # ⬅️ Optional: 'Test Time Augmentation'. Slows it down slightly but usually boosts mAP by 1-2%
+                verbose=False
+            )
             
             bbox_list = []
             category_list = []