matrice 1.0.99160__py3-none-any.whl → 1.0.99162__py3-none-any.whl

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.dist-info → matrice-1.0.99162.dist-info}/METADATA

@@ -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.dist-info → matrice-1.0.99162.dist-info}/RECORD

@@ -175,7 +175,7 @@ matrice/deploy/utils/post_processing/usecases/emergency_vehicle_detection.py,sha
 matrice/deploy/utils/post_processing/usecases/face_emotion.py,sha256=eRfqBdryB0uNoOlz_y-JMuZL1BhPWrI-odqgx_9LT7s,39132
 matrice/deploy/utils/post_processing/usecases/fashion_detection.py,sha256=f9gpzMDhIW-gyn46k9jgf8nY7YeoqAnTxGOzksabFbE,40457
 matrice/deploy/utils/post_processing/usecases/field_mapping.py,sha256=JDwYX8pd2W-waDvBh98Y_o_uchJu7wEYbFxOliA4Iq4,39822
-matrice/deploy/utils/post_processing/usecases/fire_detection.py,sha256=KU-6LEC7TNxbpASzRTUs1i44GKKe2eMvxIR8x7bFQCc,41293
+matrice/deploy/utils/post_processing/usecases/fire_detection.py,sha256=Tj4Ih19Res1-b5eA_nwjvHSDMrhIkdI3mzUlhfODhR8,41877
 matrice/deploy/utils/post_processing/usecases/flare_analysis.py,sha256=-egmS3Hs_iGOLeCMfapbkfQ04EWtZx97QRuUcDa-jMU,45340
 matrice/deploy/utils/post_processing/usecases/flower_segmentation.py,sha256=4I7qMx9Ztxg_hy9KTVX-3qBhAN-QwDt_Yigf9fFjLus,52017
 matrice/deploy/utils/post_processing/usecases/gender_detection.py,sha256=DEnCTRew6B7DtPcBQVCTtpd_IQMvMusBcu6nadUg2oM,40107
@@ -227,8 +227,8 @@ matrice/deployment/camera_manager.py,sha256=ReBZqm1CNXRImKcbcZ4uWAT3TUWkof1D28oB
 matrice/deployment/deployment.py,sha256=PLIUD-PxTaC2Zxb3Y12wUddsryV-OJetjCjLoSUh7S4,48103
 matrice/deployment/inference_pipeline.py,sha256=bXLgd29ViA7o0c7YWLFJl1otBUQfTPb61jS6VawQB0Y,37918
 matrice/deployment/streaming_gateway_manager.py,sha256=w5swGsuFVfZIdOm2ZuBHRHlRdYYJMLopLsf2gb91lQ8,20946
-matrice-1.0.99160.dist-info/licenses/LICENSE.txt,sha256=2bm9uFabQZ3Ykb_SaSU_uUbAj2-htc6WJQmS_65qD00,1073
-matrice-1.0.99160.dist-info/METADATA,sha256=hsoq2uDW91xQqRhRCM0oV6esy5qMVpnjQZrcP1Myr34,14624
-matrice-1.0.99160.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-matrice-1.0.99160.dist-info/top_level.txt,sha256=P97js8ur6o5ClRqMH3Cjoab_NqbJ6sOQ3rJmVzKBvMc,8
-matrice-1.0.99160.dist-info/RECORD,,
+matrice-1.0.99162.dist-info/licenses/LICENSE.txt,sha256=2bm9uFabQZ3Ykb_SaSU_uUbAj2-htc6WJQmS_65qD00,1073
+matrice-1.0.99162.dist-info/METADATA,sha256=7pUyVwAcw_onkhGFXYkqjFhVnNrDhIWFRlLsXV-sZok,14624
+matrice-1.0.99162.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+matrice-1.0.99162.dist-info/top_level.txt,sha256=P97js8ur6o5ClRqMH3Cjoab_NqbJ6sOQ3rJmVzKBvMc,8
+matrice-1.0.99162.dist-info/RECORD,,