matrice 1.0.99251__py3-none-any.whl → 1.0.99253__py3-none-any.whl

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

@@ -38,14 +38,14 @@ class CropWeedDetectionConfig(BaseConfig):
     smoothing_confidence_range_factor: float = 0.5
 
     # confidence thresholds
-    confidence_threshold: float = 0.6
+    confidence_threshold: float = 0.5
 
     usecase_categories: List[str] = field(
         default_factory=lambda: ['plants', 'BroWeed', 'Maize', 'NarWeed']
     )
 
     target_categories: List[str] = field(
-        default_factory=lambda: ['BroWeed', 'Maize', 'NarWeed']
+        default_factory=lambda: ['plants', 'BroWeed', 'Maize', 'NarWeed']
     )
 
     alert_config: Optional[AlertConfig] = None
@@ -75,7 +75,7 @@ class CropWeedDetectionUseCase(BaseProcessor):
         self.CASE_VERSION: Optional[str] = '1.2'
 
         # List of  categories to track
-        self.target_categories = ['BroWeed', 'Maize', 'NarWeed']
+        self.target_categories = ['plants', 'BroWeed', 'Maize', 'NarWeed']
 
         # Initialize smoothing tracker
         self.smoothing_tracker = None

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

@@ -191,6 +191,18 @@ class FireSmokeUseCase(BaseProcessor):
                     frame_number = start_frame
 
              # Step 7: alerts
+            print("-----------------------------------------------")
+            print(config.confidence_threshold)
+            print(config.fire_smoke_categories)
+            print(config.index_to_category)
+            print(config.target_categories)
+            print(config.threshold_area)
+            print(config.alert_config)
+            print(config.alert_config.count_thresholds)
+            print(config.alert_config.alert_type)
+            print(config.alert_config.alert_value)
+            print(config.alert_config.alert_incident_category)
+            print("-----------------------------------------------")
 
             alerts = self._check_alerts(fire_smoke_summary, frame_number, config, stream_info)
 
@@ -282,7 +294,11 @@ class FireSmokeUseCase(BaseProcessor):
         if hasattr(config.alert_config, 'count_thresholds') and config.alert_config.count_thresholds:
             alert_id = self._get_alert_incident_ids(self._ascending_alert_list[-1:])
 
-            for category, threshold in config.alert_config.count_thresholds.items():
+            count_thresholds = {}
+            if config.alert_config and hasattr(config.alert_config, "count_thresholds"):
+                count_thresholds = config.alert_config.count_thresholds or {}
+
+            for category, threshold in count_thresholds.items():
                 if category == "all" and total > threshold:  
                     
                     alerts.append({
@@ -360,8 +376,14 @@ class FireSmokeUseCase(BaseProcessor):
         if total > 0:
            # Calculate total bbox area
             total_area = 0.0
-        
-            for category, threshold in config.alert_config.count_thresholds.items():
+            # Safely retrieve count thresholds. If alert_config is None (e.g., when it
+            # is not provided or failed to parse) we default to an empty mapping so
+            # the subsequent logic can still execute without raising an AttributeError.
+            count_thresholds = {}
+            if config.alert_config and hasattr(config.alert_config, "count_thresholds"):
+                count_thresholds = config.alert_config.count_thresholds or {}
+
+            for category, threshold in count_thresholds.items():
                 if category in summary.get("per_category_count", {}):
                     
                     #count = summary.get("per_category_count", {})[category]
@@ -583,6 +605,26 @@ class FireSmokeUseCase(BaseProcessor):
 
 
         tracking_stats.append(tracking_stat)
+
+        if len(self.id_hit_list)==1:
+            last_ending_id = self._get_alert_incident_ids("")
+            if last_ending_id==5:
+                tracking_stats.append({
+                            "alert_type": getattr(config.alert_config, 'alert_type', ['Default']) if hasattr(config.alert_config, 'alert_type') else ['Default'],
+                            "alert_id": "alert_"+category+'_'+str(last_ending_id),
+                            "incident_category": self.CASE_TYPE,
+                            "threshold_level": 0,
+                            "ascending": False,
+                            "settings": {t: v for t, v in zip(getattr(config.alert_config, 'alert_type', ['Default']) if hasattr(config.alert_config, 'alert_type') else ['Default'],
+                                     getattr(config.alert_config, 'alert_value', ['JSON']) if hasattr(config.alert_config, 'alert_value') else ['JSON'])
+                                    }       
+                        })
+                tracking_stats.append(self.create_incident(incident_id=self.CASE_TYPE+'_'+str(last_ending_id), incident_type=self.CASE_TYPE,
+                            severity_level='info', human_text='Event Over', camera_info=camera_info, alerts=alerts, alert_settings=alert_settings,
+                            start_time=start_timestamp, end_time='Incident still active',
+                            level_settings= {"low": 3, "medium": 5, "significant":15, "critical": 30}))
+                
+        
         return tracking_stats
 
     def _generate_summary(
@@ -952,6 +994,9 @@ class FireSmokeUseCase(BaseProcessor):
             if len(self.id_hit_list)==1:
                 self.id_hit_counter+=1
                 if self.id_hit_counter>120:
+                    self.id_hit_list = ["low","medium","significant","critical","low"]
+                    self.id_hit_counter = 0
+                    self.latest_stack = None
                     return int(5)
                 if sev_level==self.latest_stack:
                     return int(5-len(self.id_hit_list))

{matrice-1.0.99251.dist-info → matrice-1.0.99253.dist-info}/METADATA

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

{matrice-1.0.99251.dist-info → matrice-1.0.99253.dist-info}/RECORD

@@ -172,7 +172,7 @@ matrice/deploy/utils/post_processing/usecases/child_monitoring.py,sha256=z3oymoq
 matrice/deploy/utils/post_processing/usecases/color_detection.py,sha256=Z8-akjy8a7f8YyiOzXu_Zi1Km30v-TRrymDqQOPpJ_8,43277
 matrice/deploy/utils/post_processing/usecases/color_map_utils.py,sha256=SP-AEVcjLmL8rxblu-ixqUJC2fqlcr7ab4hWo4Fcr_k,2677
 matrice/deploy/utils/post_processing/usecases/concrete_crack_detection.py,sha256=pxhOH_hG4hq9yytNepbGMdk2W_lTG8D1_2RAagaPBkg,40252
-matrice/deploy/utils/post_processing/usecases/crop_weed_detection.py,sha256=5_Yn40YTZfZWmqUcDUUDvoejtApfDW8qhw4I736uBj4,40498
+matrice/deploy/utils/post_processing/usecases/crop_weed_detection.py,sha256=RtKEDbkejHYeNcsSR8hCzqYinBtlGYtBsEdqaR6Kw0s,40518
 matrice/deploy/utils/post_processing/usecases/customer_service.py,sha256=UWS83qxguyAyhh8a0JF5QH9DtKxO8I-gI2BPOjLPxBw,44642
 matrice/deploy/utils/post_processing/usecases/defect_detection_products.py,sha256=blvo4wmak-wlvPSZOcmRsV1FoZSeGX_dUAX5A1WheBE,45949
 matrice/deploy/utils/post_processing/usecases/distracted_driver_detection.py,sha256=rkyYHbmcYUAfKbmmKyKxHlk47vJ_fogHWKhQjrERsok,40316
@@ -182,7 +182,7 @@ matrice/deploy/utils/post_processing/usecases/face_emotion.py,sha256=eRfqBdryB0u
 matrice/deploy/utils/post_processing/usecases/face_recognition.py,sha256=T5xAuv6b9OrkmTmoXgZs4LZ5XUsbvp9xCpeLBwdu7eI,40231
 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=RBm23nrtVHAOk49Yjb06d-xF8mGvHO_0yW9hDxQtX6k,44457
+matrice/deploy/utils/post_processing/usecases/fire_detection.py,sha256=uIeV0EJonXTBdgtS6K_Wxooncv3hzX_zajET717WNKs,47365
 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/gas_leak_detection.py,sha256=KL2ft7fXvjTas-65-QgcJm3W8KBsrwF44qibSXjfaLc,40557
@@ -243,8 +243,8 @@ matrice/deployment/camera_manager.py,sha256=e1Lc81RJP5wUWRdTgHO6tMWF9BkBdHOSVyx3
 matrice/deployment/deployment.py,sha256=HFt151eWq6iqIAMsQvurpV2WNxW6Cx_gIUVfnVy5SWE,48093
 matrice/deployment/inference_pipeline.py,sha256=6b4Mm3-qt-Zy0BeiJfFQdImOn3FzdNCY-7ET7Rp8PMk,37911
 matrice/deployment/streaming_gateway_manager.py,sha256=ifYGl3g25wyU39HwhPQyI2OgF3M6oIqKMWt8RXtMxY8,21401
-matrice-1.0.99251.dist-info/licenses/LICENSE.txt,sha256=2bm9uFabQZ3Ykb_SaSU_uUbAj2-htc6WJQmS_65qD00,1073
-matrice-1.0.99251.dist-info/METADATA,sha256=mbdD3q-rhD4szxy27qDXJXTZx6gMUzqHWIxXHfKB6S4,14624
-matrice-1.0.99251.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-matrice-1.0.99251.dist-info/top_level.txt,sha256=P97js8ur6o5ClRqMH3Cjoab_NqbJ6sOQ3rJmVzKBvMc,8
-matrice-1.0.99251.dist-info/RECORD,,
+matrice-1.0.99253.dist-info/licenses/LICENSE.txt,sha256=2bm9uFabQZ3Ykb_SaSU_uUbAj2-htc6WJQmS_65qD00,1073
+matrice-1.0.99253.dist-info/METADATA,sha256=9FdVZ2DcVsy7GKrVOFax7xL9qx3PJr8wZoEw-1D6FrY,14624
+matrice-1.0.99253.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+matrice-1.0.99253.dist-info/top_level.txt,sha256=P97js8ur6o5ClRqMH3Cjoab_NqbJ6sOQ3rJmVzKBvMc,8
+matrice-1.0.99253.dist-info/RECORD,,