Update src/localization_eval.py

src/localization_eval.py

@@ -13,7 +13,7 @@ class Localization:
     """
     Original Localization class from EvalAI
     """
-    def __init__(self, eval_bbox=True, eval_segm=False):
+    def __init__(self, eval_bbox=True, eval_segm=True):
         self.eval_bbox = eval_bbox
         self.eval_segm = eval_segm
         print("Created Localization Object")
@@ -91,16 +91,14 @@ def evaluate_submission(ground_truth_path, predictions_path):
         print(f"Required Fields Checked")
         
         # Run evaluation
-        evaluator = Localization(eval_bbox=True, eval_segm=False)
+        evaluator = Localization()
         print("Evaluation Object Instantiated")
         bbox_mAP, bbox_AP50, segm_mAP, segm_AP50 = evaluator.compute_coco_metrics(
             ground_truth_path,
             predictions_path
         )
         print(f"bbox score calculated : {bbox_mAP}")
-        # Convert mAP to percentage (0-100 scale)
-        score = round(bbox_mAP * 100, 2)
-        print("Returning Result")
+        
         return float(bbox_mAP), float(bbox_AP50), segm_mAP, segm_AP50
         
     except Exception as e: