matrice 1.0.99159__py3-none-any.whl → 1.0.99161__py3-none-any.whl

matrice/deploy/utils/post_processing/usecases/fire_detection.py

@@ -4,10 +4,11 @@ Fire and Smoke Detection use case implementation.
 This module provides a structured implementation of fire and smoke detection
 with counting, insights generation, alerting, and tracking.
 """
-from datetime import datetime, timezone
+from datetime import datetime, timezone, timedelta
 from typing import Any, Dict, List, Optional
 from dataclasses import dataclass, field
 import time
+import re
 
 from ..core.base import (
     BaseProcessor,
@@ -41,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
 
@@ -71,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]
 
 
 
@@ -86,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"
@@ -131,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:
@@ -363,10 +374,10 @@ class FireSmokeUseCase(BaseProcessor):
                     intensity_pct = min(100.0, (total_area / threshold_area) * 100)
 
                     if config.alert_config and config.alert_config.count_thresholds:
-                            if intensity_pct >= 60:
+                            if intensity_pct >= 40:
                                 level = "critical"
                                 self._ascending_alert_list.append(3)
-                            elif intensity_pct >= 40:
+                            elif intensity_pct >= 30:
                                 level = "significant"
                                 self._ascending_alert_list.append(2)
                             elif intensity_pct >= 5:
@@ -376,11 +387,11 @@ class FireSmokeUseCase(BaseProcessor):
                                 level = "low"
                                 self._ascending_alert_list.append(0)
                     else:
-                            if intensity_pct > 60:
+                            if intensity_pct > 40:
                                 level = "critical"
                                 intensity = 10.0
                                 self._ascending_alert_list.append(3)
-                            elif intensity_pct > 40:
+                            elif intensity_pct > 30:
                                 level = "significant"
                                 intensity = 9.0
                                 self._ascending_alert_list.append(2)
@@ -414,6 +425,7 @@ class FireSmokeUseCase(BaseProcessor):
                             severity_level=level, human_text=human_text, camera_info=camera_info, alerts=alerts, alert_settings=alert_settings,
                             start_time=start_timestamp, end_time=self.current_incident_end_timestamp,
                             level_settings= {"low": 1, "medium": 5, "significant":40, "critical": 60})
+                    event['duration'] = self.get_duration_seconds(start_timestamp, self.current_incident_end_timestamp)
                     incidents.append(event)
 
         else:
@@ -847,5 +859,52 @@ class FireSmokeUseCase(BaseProcessor):
         dt = datetime.fromtimestamp(timestamp, tz=timezone.utc)
         return dt.strftime('%Y:%m:%d %H:%M:%S')
 
+    def get_duration_seconds(self, start_time, end_time):
+        def parse_relative_time(t):
+            """Parse HH:MM:SS(.f) manually into timedelta"""
+            try:
+                parts = t.strip().split(":")
+                if len(parts) != 3:
+                    return None
+                hours = int(parts[0])
+                minutes = int(parts[1])
+                seconds = float(parts[2])  # works for 7.4
+                return timedelta(hours=hours, minutes=minutes, seconds=seconds)
+            except:
+                return None
+
+        def parse_time(t):
+            # Check for HH:MM:SS(.ms) format
+            if re.match(r'^\d{1,2}:\d{2}:\d{1,2}(\.\d+)?$', t):
+                return parse_relative_time(t)
+
+            # Check for full UTC format like 2025-08-01-14:23:45.123456 UTC
+            if "UTC" in t:
+                try:
+                    return datetime.strptime(t, "%Y-%m-%d-%H:%M:%S.%f UTC")
+                except ValueError:
+                    return None
+
+            return None
+
+        start_dt = parse_time(start_time)
+        end_dt = parse_time(end_time)
+
+        # Return None if invalid
+        if start_dt is None or end_dt is None:
+            print("Invalid timestamp(s). Ignoring.")
+            return 'N/A'
+
+        # If timedelta (relative time), subtract directly
+        if isinstance(start_dt, timedelta) and isinstance(end_dt, timedelta):
+            delta = end_dt - start_dt
+        elif isinstance(start_dt, datetime) and isinstance(end_dt, datetime):
+            delta = end_dt - start_dt
+        else:
+            print("Mismatched timestamp formats.")
+            return None
+
+        return delta.total_seconds()
+
 
 

{matrice-1.0.99159.dist-info → matrice-1.0.99161.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: matrice
-Version: 1.0.99159
+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.99159.dist-info → matrice-1.0.99161.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=w0vCthr1adHpMgJNSh_b9Ej2s_17q6GGN90iCsKnaLI,39467
+matrice/deploy/utils/post_processing/usecases/fire_detection.py,sha256=HKo7gyjeLiLJmzoOErT6vu7kmyPSpC5yKkXlveegJaQ,41759
 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.99159.dist-info/licenses/LICENSE.txt,sha256=2bm9uFabQZ3Ykb_SaSU_uUbAj2-htc6WJQmS_65qD00,1073
-matrice-1.0.99159.dist-info/METADATA,sha256=xbn3R0LiXnLYB5OI31ROTDxq59ztMHALTmf7oiPr1R8,14624
-matrice-1.0.99159.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-matrice-1.0.99159.dist-info/top_level.txt,sha256=P97js8ur6o5ClRqMH3Cjoab_NqbJ6sOQ3rJmVzKBvMc,8
-matrice-1.0.99159.dist-info/RECORD,,
+matrice-1.0.99161.dist-info/licenses/LICENSE.txt,sha256=2bm9uFabQZ3Ykb_SaSU_uUbAj2-htc6WJQmS_65qD00,1073
+matrice-1.0.99161.dist-info/METADATA,sha256=Tm5lKNPPf3l7BiTw05EGTX1k3lahlR2S9jRgA3CTfjo,14624
+matrice-1.0.99161.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+matrice-1.0.99161.dist-info/top_level.txt,sha256=P97js8ur6o5ClRqMH3Cjoab_NqbJ6sOQ3rJmVzKBvMc,8
+matrice-1.0.99161.dist-info/RECORD,,