fixing sahi with subset of class

shamspias · Sep 10, 2024 · ba9ad5e · ba9ad5e
1 parent 1548610
commit ba9ad5e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/utils/sahi_utils.py b/utils/sahi_utils.py
@@ -29,13 +29,14 @@ def __init__(self, debug,
 
     def load_model(self, model_path):
         """Loads a detection model based on the specified type and path."""
-        print(self.supported_classes_map)
+        # print(self.supported_classes_map)
         detection_model = AutoDetectionModel.from_pretrained(
             model_type=self.model_type,
             model=model_path,
             confidence_threshold=0.1,
             device=self.device,
             # category_mapping=self.supported_classes_map,
+            # category_remapping=self.supported_classes_map,
         )
         return detection_model