matrice 1.0.99160__tar.gz → 1.0.99161__tar.gz

{matrice-1.0.99160/src/matrice.egg-info → matrice-1.0.99161}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: matrice
-Version: 1.0.99160
+Version: 1.0.99161
 Summary: SDK for connecting to matrice.ai services
 Home-page: https://github.com/matrice-ai/python-sdk
 Author: Matrice.ai

{matrice-1.0.99160 → matrice-1.0.99161}/setup.py

@@ -14,7 +14,7 @@ long_description = (here / "README.md").read_text(encoding="utf-8")
 setup(
     name="matrice",
 
-    version = "1.0.99160",
+    version = "1.0.99161",
     
     description="SDK for connecting to matrice.ai services",
     long_description=long_description,

{matrice-1.0.99160 → matrice-1.0.99161}/src/matrice/deploy/utils/post_processing/usecases/fire_detection.py

@@ -42,6 +42,9 @@ class FireSmokeConfig(BaseConfig):
     fire_smoke_categories: List[str] = field(
         default_factory=lambda: ["Fire", "Smoke"]
     )
+    target_categories: List[str] = field(
+        default_factory=lambda: ['Fire']
+    )
 
     alert_config: Optional[AlertConfig] = None
 
@@ -72,6 +75,8 @@ class FireSmokeConfig(BaseConfig):
         self.fire_smoke_categories = [cat.lower() for cat in self.fire_smoke_categories]
         if self.index_to_category:
             self.index_to_category = {k: v.lower() for k, v in self.index_to_category.items()}
+        if self.target_categories:
+                self.target_categories = [cat.lower() for cat in self.target_categories]
 
 
 
@@ -87,6 +92,7 @@ class FireSmokeUseCase(BaseProcessor):
 
         self.smoothing_tracker = None  # Required for bbox smoothing
         self._fire_smoke_recent_history = []
+        self.target_categories=['Fire']
 
         self._ascending_alert_list: List[int] = []
         self.current_incident_end_timestamp: str = "N/A"
@@ -132,6 +138,10 @@ class FireSmokeUseCase(BaseProcessor):
                 processed_data = apply_category_mapping(processed_data, config.index_to_category)
                 self.logger.debug("Applied category mapping")
 
+            if self.target_categories:
+                processed_data = [d for d in processed_data if d.get('category') in self.target_categories]
+                self.logger.debug(f"Applied category filtering")
+
             # Step 3.5: BBox smoothing for fire/smoke
             if config.enable_smoothing:
                 if self.smoothing_tracker is None:

{matrice-1.0.99160 → matrice-1.0.99161/src/matrice.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: matrice
-Version: 1.0.99160
+Version: 1.0.99161
 Summary: SDK for connecting to matrice.ai services
 Home-page: https://github.com/matrice-ai/python-sdk
 Author: Matrice.ai