matrice-analytics 0.1.97__py3-none-any.whl → 0.1.106__py3-none-any.whl

matrice_analytics/post_processing/usecases/vehicle_monitoring_parking_lot.py

@@ -30,6 +30,10 @@ class VehicleMonitoringParkingLotConfig(BaseConfig):
     smoothing_confidence_range_factor: float = 0.5
     confidence_threshold: float = 0.6
 
+    # Class Aggregation: Configuration parameters
+    enable_class_aggregation: bool = True
+    class_aggregation_window_size: int = 30  # 30 frames ≈ 1 second at 30 FPS
+
     #JBK_720_GATE POLYGON = [[86, 328], [844, 317], [1277, 520], [1273, 707], [125, 713]]
     zone_config: Optional[Dict[str, List[List[float]]]] = None #field(
 #     default_factory=lambda: {
@@ -157,9 +161,22 @@ class VehicleMonitoringParkingLotUseCase(BaseProcessor):
             from ..advanced_tracker import AdvancedTracker
             from ..advanced_tracker.config import TrackerConfig
             if self.tracker is None:
-                tracker_config = TrackerConfig()
+                tracker_config = TrackerConfig(
+                    # CLASS AGGREGATION: Map from use case config
+                    enable_class_aggregation=config.enable_class_aggregation,
+                    class_aggregation_window_size=config.class_aggregation_window_size
+                )
                 self.tracker = AdvancedTracker(tracker_config)
                 self.logger.info("Initialized AdvancedTracker for Vehicle Monitoring Parking Lot use case")
+
+                if config.enable_class_aggregation:
+                    self.logger.info(
+                        f"AdvancedTracker initialized with class aggregation "
+                        f"(window_size={config.class_aggregation_window_size})"
+                    )
+                else:
+                    self.logger.info("AdvancedTracker initialized without class aggregation")
+            
             processed_data = self.tracker.update(processed_data)
         except Exception as e:
             self.logger.warning(f"AdvancedTracker failed: {e}")

matrice_analytics/post_processing/utils/__init__.py

@@ -58,6 +58,10 @@ from .smoothing_utils import (
     create_default_smoothing_config
 )
 
+from .agnostic_nms import (
+    AgnosticNMS
+)
+
 # from .color_utils import (
 #     extract_major_colors
 # )
@@ -107,6 +111,7 @@ __all__ = [
     'remove_duplicate_detections',
     'apply_category_mapping',
     'filter_by_area',
+    'AgnosticNMS',
     
     # Counting utilities
     'count_objects_by_category',