matrice 1.0.99160__tar.gz → 1.0.99162__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: matrice
-Version: 1.0.99160
+Version: 1.0.99162
 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.99162}/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.99162",
     
     description="SDK for connecting to matrice.ai services",
     long_description=long_description,

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

@@ -40,7 +40,10 @@ class FireSmokeConfig(BaseConfig):
 
     # Only fire and smoke categories included here (exclude normal)
     fire_smoke_categories: List[str] = field(
-        default_factory=lambda: ["Fire", "Smoke"]
+        default_factory=lambda: ["fire", "smoke"]
+    )
+    target_categories: List[str] = field(
+        default_factory=lambda: ['fire']
     )
 
     alert_config: Optional[AlertConfig] = None
@@ -51,8 +54,8 @@ class FireSmokeConfig(BaseConfig):
     # Map only fire and smoke; ignore normal (index 1 not included)
     index_to_category: Optional[Dict[int, str]] = field(
         default_factory=lambda: {
-            0: "Fire",
-            1: "Smoke",
+            0: "fire",
+            1: "smoke",
         }
     )
 
@@ -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').lower() 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:
@@ -177,6 +187,8 @@ class FireSmokeUseCase(BaseProcessor):
                     frame_number = start_frame
 
              # Step 7: alerts
+            print("-----------ALERT_PROCESSING_IPP-----------")
+            print(fire_smoke_summary)
             alerts = self._check_alerts(fire_smoke_summary, frame_number, config, stream_info)
 
 
@@ -587,7 +599,7 @@ class FireSmokeUseCase(BaseProcessor):
             ]
             counts = {}
             for det in detections:
-                cat = det.get('category', 'unknown')
+                cat = det.get('category', 'unknown').lower()
                 counts[cat] = counts.get(cat, 0) + 1
             
 
@@ -715,7 +727,7 @@ class FireSmokeUseCase(BaseProcessor):
                 "fire_smoke_categories": {
                     "type": "array",
                     "items": {"type": "string"},
-                    "default": ["Fire", "Smoke"],
+                    "default": ["fire", "smoke"],
                     "description": "Category names that represent fire and smoke",
                 },
                 "index_to_category": {
@@ -743,8 +755,8 @@ class FireSmokeUseCase(BaseProcessor):
         defaults = {
             "category": self.category,
             "usecase": self.name,
-            "confidence_threshold": 0.5,
-            "fire_smoke_categories": ["Fire", "Smoke"],
+            "confidence_threshold": 0.3,
+            "fire_smoke_categories": ["fire", "smoke"],
         }
         defaults.update(overrides)
         return FireSmokeConfig(**defaults)

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

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