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

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

@@ -562,7 +562,7 @@ class ProximityUseCase(BaseProcessor):
                     human_text_lines.append("Alerts: None")
                 human_text = "\n".join(human_text_lines)  
         else:      
-            human_text = self._generate_human_text_for_tracking(total_people, total_unique_count, config, frame_id, alerts, stream_info)
+            human_text = self._generate_human_text_for_tracking(total_people, detections, total_unique_count, config, frame_id, alerts, stream_info)
         
         # Create high precision timestamps for input_timestamp and reset_timestamp
         high_precision_start_timestamp = self._get_current_timestamp_str(stream_info, precision=True, frame_id=frame_id)
@@ -571,12 +571,12 @@ class ProximityUseCase(BaseProcessor):
         tracking_stat = self.create_tracking_stats(
             total_counts, current_counts, detections, human_text, camera_info, alerts, alert_settings, start_time=high_precision_start_timestamp, reset_time=high_precision_reset_timestamp
         )
-        
+        print(tracking_stat)
         return [tracking_stat]
     
     def _check_proximity(self, detections):
         """Check if any two detections are within proximity threshold."""
-        proximity_threshold = 0.5
+        proximity_threshold = 400
         for i in range(len(detections)):
             for j in range(i + 1, len(detections)):
                 bbox1 = detections[i].get("bounding_box", {})
@@ -589,11 +589,12 @@ class ProximityUseCase(BaseProcessor):
                 center2 = get_bbox_center(bbox2)
                 
                 if center1 and center2:
-                    distance = ((center1[0] - center2[0]) ** 2 + (center1[1] - center2[1]) ** 2) ** 0.5
+                    distance = abs((center1[0] - center2[0]) + (center1[1] - center2[1]))
                     if distance < proximity_threshold:
                         return True
+            return False
     
-    def _generate_human_text_for_tracking(self, total_people: int, total_unique_count: int, config: ProximityConfig, frame_id: str, alerts:Any=[], stream_info: Optional[Dict[str, Any]] = None) -> str:
+    def _generate_human_text_for_tracking(self, total_people: int, detections, total_unique_count: int, config: ProximityConfig, frame_id: str, alerts:Any=[], stream_info: Optional[Dict[str, Any]] = None) -> str:
         """Generate human-readable text for tracking stats in old format."""
         from datetime import datetime, timezone
         
@@ -602,7 +603,9 @@ class ProximityUseCase(BaseProcessor):
         start_timestamp = self._get_start_timestamp_str(stream_info, precision=True)
 
         human_text_lines.append(f"CURRENT FRAME @ {current_timestamp}:")
-        human_text_lines.append(f"\t- People Detected: {total_people}")
+        if self._check_proximity(detections):
+            human_text_lines.append("\t- Proximity Alert: People are within proximity threshold!")
+        # human_text_lines.append(f"\t- People Detected: {total_people}")
 
         human_text_lines.append("")
         if total_unique_count > 0:

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

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

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

@@ -204,7 +204,7 @@ matrice/deploy/utils/post_processing/usecases/plaque_segmentation_img.py,sha256=
 matrice/deploy/utils/post_processing/usecases/pothole_segmentation.py,sha256=jXTb8ZqInp5xJ-O3Zp3zQBiryFVD0-WBbhW6Kux_NDo,44905
 matrice/deploy/utils/post_processing/usecases/ppe_compliance.py,sha256=G9P9j9E9nfNJInHJxmK1Lb4daFBlG5hq0aqotTLvFFE,30146
 matrice/deploy/utils/post_processing/usecases/price_tag_detection.py,sha256=09Tp6MGAHh95s-NSAp-4WC9iCc20sajWApuUBAvgXiQ,39880
-matrice/deploy/utils/post_processing/usecases/proximity_detection.py,sha256=eLs_C2-XeNH-XwTOKeXxDZHafvwmYQqSkf8pyTO3Jl0,77438
+matrice/deploy/utils/post_processing/usecases/proximity_detection.py,sha256=yOIDAP8B484cvHIiLUNqwDBDsu3eZIV_61FAMD2opAo,77640
 matrice/deploy/utils/post_processing/usecases/road_lane_detection.py,sha256=V_KxwBtAHSNkyoH8sXw-U-P3J8ToXtX3ncc69gn6Tds,31591
 matrice/deploy/utils/post_processing/usecases/road_traffic_density.py,sha256=YiHQ0kKhXglagHPvygywxMqZAw8s0WharrBQqLQj2q4,40311
 matrice/deploy/utils/post_processing/usecases/road_view_segmentation.py,sha256=BcBbOOg5622KuvzKrzs9cJW1wkRoIIcOab0N7BONQKQ,44986
@@ -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.99249.dist-info/licenses/LICENSE.txt,sha256=2bm9uFabQZ3Ykb_SaSU_uUbAj2-htc6WJQmS_65qD00,1073
-matrice-1.0.99249.dist-info/METADATA,sha256=lqoVX_xeHrWV65FNXH7oHyADPM6oPg-IZyThL0y_Cgs,14624
-matrice-1.0.99249.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-matrice-1.0.99249.dist-info/top_level.txt,sha256=P97js8ur6o5ClRqMH3Cjoab_NqbJ6sOQ3rJmVzKBvMc,8
-matrice-1.0.99249.dist-info/RECORD,,
+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,,