matrice 1.0.98848__py3-none-any.whl → 1.0.98849__py3-none-any.whl

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

@@ -399,44 +399,59 @@ class PPEComplianceUseCase(BaseProcessor):
             frame_events.append(alert_event)
         return events
 
-
-    def _generate_tracking_stats(self, counting_summary: Dict, insights: List[str], summary: str, config: PPEComplianceConfig, frame_number: Optional[int] = None, stream_info: Optional[Dict[str, Any]] = None) -> List[Dict]:
+    def _generate_tracking_stats(
+            self,
+            counting_summary: Dict,
+            insights: List[str],
+            summary: str,
+            config: PPEComplianceConfig,
+            frame_number: Optional[int] = None,
+            stream_info: Optional[Dict[str, Any]] = None
+    ) -> List[Dict]:
         """Generate structured tracking stats for the output format with frame-based keys, including track_ids_info."""
+        from datetime import datetime, timezone
+
         frame_key = str(frame_number) if frame_number is not None else "current_frame"
         tracking_stats = [{frame_key: []}]
         frame_tracking_stats = tracking_stats[0][frame_key]
+
         total_violations = counting_summary.get("total_count", 0)
         per_cat = counting_summary.get("per_category_count", {})
         cumulative = counting_summary.get("total_violation_counts", {})
         cumulative_total = sum(cumulative.values()) if cumulative else 0
-        # Always get track ID info even if 0 detections — to ensure consistency
+
         track_ids_info = self._get_track_ids_info(counting_summary.get("detections", []))
-        # Get formatted timestamps
+
         current_timestamp = self._get_current_timestamp_str(stream_info)
         start_timestamp = self._get_start_timestamp_str(stream_info)
-        # Build human-readable summary string in requested format
+
         human_text_lines = []
+
         # CURRENT FRAME section
         human_text_lines.append(f"CURRENT FRAME @ {current_timestamp}:")
         if total_violations > 0:
-            human_text_lines.append(f"    - PPE Violations Detected: {total_violations}")
+            human_text_lines.append(f"\t- PPE Violations Detected: {total_violations}")
             for cat in ["NO-Hardhat", "NO-Mask", "NO-Safety Vest"]:
                 count = per_cat.get(cat, 0)
                 if count > 0:
                     label = self.CATEGORY_DISPLAY.get(cat, cat).replace(" Violations", "")
-                    human_text_lines.append(f"        - {label}: {count}")
+                    human_text_lines.append(f"\t\t- {label}: {count}")
         else:
-            human_text_lines.append("    - No PPE violations detected")
-        human_text_lines.append("")  # Empty line for spacing
+            human_text_lines.append("\t- No PPE violations detected")
+
+        human_text_lines.append("")  # spacing
+
         # TOTAL SINCE section
         human_text_lines.append(f"TOTAL SINCE {start_timestamp}:")
-        human_text_lines.append(f"    - Total PPE Violations Detected: {cumulative_total}")
+        human_text_lines.append(f"\t- Total PPE Violations Detected: {cumulative_total}")
         for cat in ["NO-Hardhat", "NO-Mask", "NO-Safety Vest"]:
             count = cumulative.get(cat, 0)
             if count > 0:
                 label = self.CATEGORY_DISPLAY.get(cat, cat).replace(" Violations", "")
-                human_text_lines.append(f"        - {label}: {count}")
+                human_text_lines.append(f"\t\t- {label}: {count}")
+
         human_text = "\n".join(human_text_lines)
+
         tracking_stat = {
             "type": "ppe_tracking",
             "category": "ppe",
@@ -449,11 +464,10 @@ class PPEComplianceUseCase(BaseProcessor):
             "global_frame_offset": getattr(self, '_global_frame_offset', 0),
             "local_frame_id": frame_key
         }
+
         frame_tracking_stats.append(tracking_stat)
         return tracking_stats
 
-
-
     def _count_categories(self, detections: list, config: PPEComplianceConfig) -> dict:
         """
         Count the number of detections per category and return a summary dict.

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

@@ -547,40 +547,47 @@ class VehicleMonitoringUseCase(BaseProcessor):
 
         return events
 
-    def _generate_tracking_stats(self, counting_summary: Dict, insights: List[str], summary: str,
-                                 config: VehicleMonitoringConfig, frame_number: Optional[int] = None,
-                                 stream_info: Optional[Dict[str, Any]] = None) -> List[Dict]:
+    def _generate_tracking_stats(
+            self,
+            counting_summary: Dict,
+            insights: List[str],
+            summary: str,
+            config: VehicleMonitoringConfig,
+            frame_number: Optional[int] = None,
+            stream_info: Optional[Dict[str, Any]] = None
+    ) -> List[Dict]:
         """Generate structured tracking stats for the output format with frame-based keys, including track_ids_info."""
+        from datetime import datetime, timezone
+
         frame_key = str(frame_number) if frame_number is not None else "current_frame"
         tracking_stats = [{frame_key: []}]
         frame_tracking_stats = tracking_stats[0][frame_key]
+
         total_vehicles = counting_summary.get("total_count", 0)
         total_vehicle_counts = counting_summary.get("total_vehicle_counts", {})
         cumulative_total = sum(total_vehicle_counts.values()) if total_vehicle_counts else 0
-        # Add detailed track_ids_info (like people_counting)
+
         track_ids_info = self._get_track_ids_info(counting_summary.get("detections", []))
 
-        # Get formatted timestamps
         current_timestamp = self._get_current_timestamp_str(stream_info)
         start_timestamp = self._get_start_timestamp_str(stream_info)
 
-        # Build human-readable summary string in new format
         human_text_lines = []
 
         # CURRENT FRAME section
         human_text_lines.append(f"CURRENT FRAME @ {current_timestamp}:")
         if total_vehicles > 0:
-            human_text_lines.append(f"    - Vehicles Detected: {total_vehicles}")
+            human_text_lines.append(f"\t- Vehicles Detected: {total_vehicles}")
         else:
-            human_text_lines.append("    - No Vehicles detected")
+            human_text_lines.append(f"\t- No vehicles detected")
 
-        human_text_lines.append("")  # Empty line for spacing
+        human_text_lines.append("")  # spacing
 
         # TOTAL SINCE section
         human_text_lines.append(f"TOTAL SINCE {start_timestamp}:")
-        human_text_lines.append(f"    - Total Vehicles Detected: {cumulative_total}")
+        human_text_lines.append(f"\t- Total Vehicles Detected: {cumulative_total}")
         for cat, count in total_vehicle_counts.items():
-            human_text_lines.append(f"        • {cat.capitalize()}: {count}")
+            human_text_lines.append(f"\t\t- {cat.capitalize()}: {count}")
 
         human_text = "\n".join(human_text_lines)
 
@@ -596,8 +603,8 @@ class VehicleMonitoringUseCase(BaseProcessor):
             "global_frame_offset": getattr(self, '_global_frame_offset', 0),
             "local_frame_id": frame_key
         }
-        frame_tracking_stats.append(tracking_stat)
 
+        frame_tracking_stats.append(tracking_stat)
         return tracking_stats
 
     def _count_categories(self, detections: list, config: VehicleMonitoringConfig) -> dict:

{matrice-1.0.98848.dist-info → matrice-1.0.98849.dist-info}/METADATA

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

{matrice-1.0.98848.dist-info → matrice-1.0.98849.dist-info}/RECORD

@@ -122,8 +122,8 @@ matrice/deploy/utils/post_processing/usecases/fire_detection.py,sha256=rLaUvU5yg
 matrice/deploy/utils/post_processing/usecases/flare_analysis.py,sha256=_pPKa16ax4-V-iOHkdLp2-sYA6uJNAm7Z-75Fnv6vro,48674
 matrice/deploy/utils/post_processing/usecases/license_plate_detection.py,sha256=zxFI8iakFUCBm64P83l380CEiuGGOPNzhZhHNpAaxIc,26902
 matrice/deploy/utils/post_processing/usecases/people_counting.py,sha256=QSvq2zwFotp0o_T15EqFePAkukIyACleBZCf4fy7dHM,77973
-matrice/deploy/utils/post_processing/usecases/ppe_compliance.py,sha256=Jg_ZzQ0qzQikqEr-mq-xCX_s10CmACaDGZAOcwaqvjA,27712
-matrice/deploy/utils/post_processing/usecases/vehicle_monitoring.py,sha256=ZqwZYY31AZg07OpnnB1bfsS0EkwUfQE_o-tQlbnZJZ4,39431
+matrice/deploy/utils/post_processing/usecases/ppe_compliance.py,sha256=4CsmjEbwmYlvQgC0kEhjEQlkwkgwuD4AG2NKwvfWqSs,27646
+matrice/deploy/utils/post_processing/usecases/vehicle_monitoring.py,sha256=C6vLwxyjGKxuLQ9N2oaFCwQnoxKV1YsgqMA7hNflTR4,39324
 matrice/deploy/utils/post_processing/utils/__init__.py,sha256=A49ksdXL7gRwBbIUwnU2ueFDGA67qVnEW_9lItOibtk,3626
 matrice/deploy/utils/post_processing/utils/advanced_counting_utils.py,sha256=D6jlZNRCfPtfG8COv3AMCbCfZf4_DK9rFhwzVJEYjpg,19152
 matrice/deploy/utils/post_processing/utils/advanced_helper_utils.py,sha256=W8mDqJTpg98YJgWYBod0rZUNbR4bmvYMeWAGASs14_s,11624
@@ -137,8 +137,8 @@ matrice/deploy/utils/post_processing/utils/format_utils.py,sha256=ce2VpQicTYKfxu
 matrice/deploy/utils/post_processing/utils/geometry_utils.py,sha256=BWfdM6RsdJTTLR1GqkWfdwpjMEjTCJyuBxA4zVGKdfk,9623
 matrice/deploy/utils/post_processing/utils/smoothing_utils.py,sha256=gQPykV1HGh-IMMz_3fumgtrhpyCRgXMS1hOEDcLRMO0,12542
 matrice/deploy/utils/post_processing/utils/tracking_utils.py,sha256=rWxuotnJ3VLMHIBOud2KLcu4yZfDp7hVPWUtNAq_2xw,8288
-matrice-1.0.98848.dist-info/licenses/LICENSE.txt,sha256=2bm9uFabQZ3Ykb_SaSU_uUbAj2-htc6WJQmS_65qD00,1073
-matrice-1.0.98848.dist-info/METADATA,sha256=r1CvrLyr0z_IDiB2AQEGI6bTrgtMTmyUqj8PZ_ZoIuQ,14624
-matrice-1.0.98848.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-matrice-1.0.98848.dist-info/top_level.txt,sha256=P97js8ur6o5ClRqMH3Cjoab_NqbJ6sOQ3rJmVzKBvMc,8
-matrice-1.0.98848.dist-info/RECORD,,
+matrice-1.0.98849.dist-info/licenses/LICENSE.txt,sha256=2bm9uFabQZ3Ykb_SaSU_uUbAj2-htc6WJQmS_65qD00,1073
+matrice-1.0.98849.dist-info/METADATA,sha256=pBRtfgYqdmoS4hZ6RtJoGcjRrQlxJxtw1V0lreQhIzI,14624
+matrice-1.0.98849.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+matrice-1.0.98849.dist-info/top_level.txt,sha256=P97js8ur6o5ClRqMH3Cjoab_NqbJ6sOQ3rJmVzKBvMc,8
+matrice-1.0.98849.dist-info/RECORD,,