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

matrice_analytics/post_processing/__init__.py

@@ -85,7 +85,7 @@ from .usecases.leaf_disease import LeafDiseaseDetectionConfig, LeafDiseaseDetect
 from .usecases.parking import ParkingConfig
 from .usecases.abandoned_object_detection import AbandonedObjectConfig
 from .usecases.footfall import FootFallConfig
-
+from .usecases.vehicle_monitoring import VehicleMonitoringConfig
 
 from .usecases.weld_defect_detection import WeldDefectConfig
 from .usecases.weapon_detection import WeaponDetectionConfig
@@ -130,6 +130,8 @@ from .usecases.underground_pipeline_defect_detection import UndergroundPipelineD
 from .usecases.suspicious_activity_detection import SusActivityConfig, SusActivityUseCase
 from .usecases.natural_disaster import NaturalDisasterConfig, NaturalDisasterUseCase
 from .usecases.footfall import FootFallUseCase
+from .usecases.vehicle_monitoring_parking_lot import VehicleMonitoringParkingLotUseCase, VehicleMonitoringParkingLotConfig
+from .usecases.vehicle_monitoring_drone_view import VehicleMonitoringDroneViewUseCase, VehicleMonitoringDroneViewConfig
 
 #Put all IMAGE based usecases here
 from .usecases.blood_cancer_detection_img import BloodCancerDetectionConfig, BloodCancerDetectionUseCase
@@ -208,6 +210,9 @@ from .usecases import (
     SusActivityUseCase,
     NaturalDisasterUseCase,
     FootFallUseCase,
+    VehicleMonitoringParkingLotUseCase,
+    VehicleMonitoringDroneViewUseCase,
+
     #Put all IMAGE based usecases here
     BloodCancerDetectionUseCase,
     SkinCancerClassificationUseCase,
@@ -289,6 +294,8 @@ _underground_pipeline_defect = UndergroundPipelineDefectUseCase()
 _suspicious_activity_detection = SusActivityUseCase()
 _natural_disaster = NaturalDisasterUseCase()
 _footfall = FootFallUseCase()
+_vehicle_monitoring_parking_lot = VehicleMonitoringParkingLotUseCase()
+_vehicle_monitoring_drone_view = VehicleMonitoringDroneViewUseCase()
 
 # Face recognition with embeddings
 _face_recognition = FaceRecognitionEmbeddingUseCase()
@@ -374,6 +381,8 @@ registry.register_use_case(_underground_pipeline_defect.category, _underground_p
 registry.register_use_case(_suspicious_activity_detection.category, _suspicious_activity_detection.name, SusActivityUseCase)
 registry.register_use_case(_natural_disaster.category, _natural_disaster.name, NaturalDisasterUseCase)
 registry.register_use_case(_footfall.category, _footfall.name, FaceEmotionUseCase)
+registry.register_use_case(_vehicle_monitoring_parking_lot.category, _vehicle_monitoring_parking_lot.name, VehicleMonitoringParkingLotUseCase)
+registry.register_use_case(_vehicle_monitoring_drone_view.category, _vehicle_monitoring_drone_view.name, VehicleMonitoringDroneViewUseCase)
 
 #Put all IMAGE based usecases here
 registry.register_use_case(_blood_cancer_detection.category, _blood_cancer_detection.name, BloodCancerDetectionUseCase)
@@ -580,6 +589,8 @@ __all__ = [
     'NaturalDisasterConfig',
     'VehiclePeopleDroneMonitoringConfig',
     'FootFallConfig',
+    'VehicleMonitoringParkingLotConfig',
+    'VehicleMonitoringDroneViewConfig',
     #Put all IMAGE based usecase CONFIGS here
     'BloodCancerDetectionConfig',
     'SkinCancerClassificationConfig',
@@ -654,6 +665,8 @@ __all__ = [
     'SusActivityUseCase',
     'NaturalDisasterUseCase',
     'FootFallUseCase',
+    'VehicleMonitoringParkingLotUseCase',
+    'VehicleMonitoringDroneViewUseCase',
 
     #Put all IMAGE based usecases here
     'BloodCancerDetectionUseCase',

matrice_analytics/post_processing/config.py

@@ -65,7 +65,9 @@ APP_NAME_TO_USECASE = {
     "underground_pipeline_defect" : "underground_pipeline_defect",
     "suspicious_activity_detection": "suspicious_activity_detection",
     "natural_disaster_detection": "natural_disaster_detection",
-    "Foot Fall": "footfall"
+    "Foot Fall": "footfall",
+    "vehicle_monitoring_parking_lot": "vehicle_monitoring_parking_lot",
+    "vehicle_monitoring_drone_view": "vehicle_monitoring_drone_view",
 }
 
 APP_NAME_TO_CATEGORY = {
@@ -136,7 +138,9 @@ APP_NAME_TO_CATEGORY = {
     "underground_pipeline_defect" : "general",
     "suspicious_activity_detection": "security",
     "natural_disaster_detection": "environmental",
-    "Foot Fall": "retail"
+    "Foot Fall": "retail",
+    "vehicle_monitoring_parking_lot": "traffic",
+    "vehicle_monitoring_drone_view": "traffic",
 }
 
 def get_usecase_from_app_name(app_name: str) -> str:

matrice_analytics/post_processing/core/config.py

@@ -907,6 +907,8 @@ class ConfigManager:
             'suspicious_activity_detection': None,
             'natural_disaster_detection': None,
             'footfall': None,
+            'vehicle_monitoring_parking_lot': None,
+            'vehicle_monitoring_drone_view': None,
 
             #Put all image based usecases here::
             'blood_cancer_detection_img': None,
@@ -1420,6 +1422,22 @@ class ConfigManager:
         except ImportError:
             return None
         
+    def vehicle_monitoring_parking_lot_config_class(self):
+        """Register a configuration class for a use case."""
+        try:
+            from ..usecases.vehicle_monitoring_parking_lot import VehicleMonitoringParkingLotConfig
+            return VehicleMonitoringParkingLotConfig
+        except ImportError:
+            return None
+        
+    def vehicle_monitoring_drone_view_config_class(self):
+        """Register a configuration class for a use case."""
+        try:
+            from ..usecases.vehicle_monitoring_drone_view import VehicleMonitoringDroneViewConfig
+            return VehicleMonitoringDroneViewConfig
+        except ImportError:
+            return None
+        
     #put all image based usecases here::
     def blood_cancer_detection_config_class(self):
         """Register a configuration class for a use case."""
@@ -2713,6 +2731,38 @@ class ConfigManager:
                 alert_config=alert_config,
                 **kwargs
             )
+
+        elif usecase == "vehicle_monitoring_parking_lot":
+            # Import here to avoid circular import
+            from ..usecases.vehicle_monitoring_parking_lot import VehicleMonitoringParkingLotConfig
+
+            # Handle nested configurations
+            alert_config = kwargs.pop("alert_config", None)
+            if alert_config and isinstance(alert_config, dict):
+                alert_config = AlertConfig(**alert_config)
+
+            config = VehicleMonitoringParkingLotConfig(
+                category=category or "traffic",
+                usecase=usecase,
+                alert_config=alert_config,
+                **kwargs
+            )
+
+        elif usecase == "vehicle_monitoring_drone_view":
+            # Import here to avoid circular import
+            from ..usecases.vehicle_monitoring_drone_view import VehicleMonitoringDroneViewConfig
+
+            # Handle nested configurations
+            alert_config = kwargs.pop("alert_config", None)
+            if alert_config and isinstance(alert_config, dict):
+                alert_config = AlertConfig(**alert_config)
+
+            config = VehicleMonitoringDroneViewConfig(
+                category=category or "traffic",
+                usecase=usecase,
+                alert_config=alert_config,
+                **kwargs
+            )
         
         #Add IMAGE based usecases here::
         elif usecase == "blood_cancer_detection_img":
@@ -3271,6 +3321,18 @@ class ConfigManager:
             default_config = FootFallConfig()
             return default_config.to_dict()
         
+        elif usecase == "vehicle_monitoring_parking_lot":
+            # Import here to avoid circular import
+            from ..usecases.vehicle_monitoring_parking_lot import VehicleMonitoringParkingLotConfig
+            default_config = VehicleMonitoringParkingLotConfig()
+            return default_config.to_dict()
+        
+        elif usecase == "vehicle_monitoring_drone_view":
+            # Import here to avoid circular import
+            from ..usecases.vehicle_monitoring_drone_view import VehicleMonitoringDroneViewConfig
+            default_config = VehicleMonitoringDroneViewConfig()
+            return default_config.to_dict()
+        
         
         elif usecase == "underground_pipeline_defect":
             # Import here to avoid circular import

matrice_analytics/post_processing/post_processor.py

@@ -100,6 +100,8 @@ from .usecases import (
     SusActivityUseCase,
     NaturalDisasterUseCase,
     FootFallUseCase,
+    VehicleMonitoringParkingLotUseCase,
+    VehicleMonitoringDroneViewUseCase,
     # Put all IMAGE based usecases here
     BloodCancerDetectionUseCase,
     SkinCancerClassificationUseCase,
@@ -573,6 +575,12 @@ class PostProcessor:
         registry.register_use_case(
             "retail", "footfall", FootFallUseCase
         )
+        registry.register_use_case(
+            "traffic", "vehicle_monitoring_parking_lot", VehicleMonitoringParkingLotUseCase
+        )
+        registry.register_use_case(
+            "traffic", "vehicle_monitoring_drone_view", VehicleMonitoringDroneViewUseCase
+        )
 
         # Put all IMAGE based usecases here
         registry.register_use_case(

matrice_analytics/post_processing/usecases/__init__.py

@@ -86,7 +86,8 @@ from .underground_pipeline_defect_detection import UndergroundPipelineDefectConf
 from .suspicious_activity_detection import SusActivityConfig, SusActivityUseCase
 from .natural_disaster import NaturalDisasterConfig, NaturalDisasterUseCase
 from .footfall import FootFallConfig, FootFallUseCase
-
+from .vehicle_monitoring_parking_lot import VehicleMonitoringParkingLotUseCase, VehicleMonitoringParkingLotConfig
+from .vehicle_monitoring_drone_view import VehicleMonitoringDroneViewUseCase, VehicleMonitoringDroneViewConfig
 
 #Put all IMAGE based usecases here
 from .blood_cancer_detection_img import BloodCancerDetectionConfig, BloodCancerDetectionUseCase
@@ -175,6 +176,8 @@ __all__ = [
     'SusActivityUseCase',
     'NaturalDisasterUseCase',
     'FootFallUseCase',
+    'VehicleMonitoringParkingLotUseCase',
+    'VehicleMonitoringDroneViewUseCase',
 
     #Put all IMAGE based usecases here
     'BloodCancerDetectionUseCase',
@@ -258,6 +261,9 @@ __all__ = [
     'SusActivityConfig',
     'NaturalDisasterConfig',
     'FootFallConfig',
+    'VehicleMonitoringParkingLotConfig',
+    'VehicleMonitoringDroneViewConfig',
+
     #Put all IMAGE based usecase CONFIGS here
     'BloodCancerDetectionConfig',
     'SkinCancerClassificationConfig',

matrice_analytics/post_processing/usecases/vehicle_monitoring.py

@@ -40,54 +40,35 @@ class VehicleMonitoringConfig(BaseConfig):
 # )
     usecase_categories: List[str] = field(
         default_factory=lambda: [
-            "person", "bicycle", "car", "motorcycle", "airplane", "bus", "train", "truck", "boat",
-            "traffic light", "fire hydrant", "stop sign", "parking meter", "bench", "bird", "cat", "dog",
-            "horse", "sheep", "cow", "elephant", "bear", "zebra", "giraffe", "backpack", "umbrella",
-            "handbag", "tie", "suitcase", "frisbee", "skis", "snowboard", "sports ball", "kite",
-            "baseball bat", "baseball glove", "skateboard", "surfboard", "tennis racket", "bottle",
-            "wine glass", "cup", "fork", "knife", "spoon", "bowl", "banana", "apple", "sandwich",
-            "orange", "broccoli", "carrot", "hot dog", "pizza", "donut", "cake", "chair", "couch",
-            "potted plant", "bed", "dining table", "toilet", "tv", "laptop", "mouse", "remote",
-            "keyboard", "cell phone", "microwave", "oven", "toaster", "sink", "refrigerator", "book",
-            "clock", "vase", "scissors", "teddy bear", "hair drier", "toothbrush"
+            'bicycle', 'motorcycle', 'car', 'van', 'bus', 'truck' 
         ]
     )
     target_categories: List[str] = field(
         default_factory=lambda: [
-            'car', 'bicycle', 'bus','motorcycle']
+            'bicycle', 'motorcycle', 'car', 'van', 'bus', 'truck' 
+        ]
     )
     alert_config: Optional[AlertConfig] = None
     index_to_category: Optional[Dict[int, str]] = field(
         default_factory=lambda: {
-                0: "person", 1: "bicycle", 2: "car", 3: "motorcycle", 4: "airplane", 5: "bus",
-                6: "train", 7: "truck", 8: "boat", 9: "traffic light", 10: "fire hydrant",
-                11: "stop sign", 12: "parking meter", 13: "bench", 14: "bird", 15: "cat",
-                16: "dog", 17: "horse", 18: "sheep", 19: "cow", 20: "elephant", 21: "bear",
-                22: "zebra", 23: "giraffe", 24: "backpack", 25: "umbrella", 26: "handbag",
-                27: "tie", 28: "suitcase", 29: "frisbee", 30: "skis", 31: "snowboard",
-                32: "sports ball", 33: "kite", 34: "baseball bat", 35: "baseball glove",
-                36: "skateboard", 37: "surfboard", 38: "tennis racket", 39: "bottle",
-                40: "wine glass", 41: "cup", 42: "fork", 43: "knife", 44: "spoon", 45: "bowl",
-                46: "banana", 47: "apple", 48: "sandwich", 49: "orange", 50: "broccoli",
-                51: "carrot", 52: "hot dog", 53: "pizza", 54: "donut", 55: "cake", 56: "chair",
-                57: "couch", 58: "potted plant", 59: "bed", 60: "dining table", 61: "toilet",
-                62: "tv", 63: "laptop", 64: "mouse", 65: "remote", 66: "keyboard",
-                67: "cell phone", 68: "microwave", 69: "oven", 70: "toaster", 71: "sink",
-                72: "refrigerator", 73: "book", 74: "clock", 75: "vase", 76: "scissors",
-                77: "teddy bear", 78: "hair drier", 79: "toothbrush"
-            }
+            0: "bicycle",
+            1: "motorcycle", 
+            2: "car",
+            3: "van",
+            4: "bus",
+            5: "truck"
+        }
     )
 
 class VehicleMonitoringUseCase(BaseProcessor):
     CATEGORY_DISPLAY = {
         # Focus on vehicle-related COCO classes
         "bicycle": "Bicycle",
-        "car": "Car",
         "motorcycle": "Motorcycle",
+        "car": "Car",
+        "van": "Van",    
         "bus": "Bus",
         "truck": "Truck",
-        "train": "Train",
-        "boat": "Boat"
     }
 
     def __init__(self):
@@ -95,7 +76,7 @@ class VehicleMonitoringUseCase(BaseProcessor):
         self.category = "traffic"
         self.CASE_TYPE: Optional[str] = 'vehicle_monitoring'
         self.CASE_VERSION: Optional[str] = '1.0'
-        self.target_categories = ['car', 'bicycle', 'bus', 'truck', 'motorcycle']
+        self.target_categories = ['bicycle', 'motorcycle', 'car', 'van', 'bus', 'truck' ]
         self.smoothing_tracker = None
         self.tracker = None
         self._total_frame_counter = 0
@@ -141,6 +122,7 @@ class VehicleMonitoringUseCase(BaseProcessor):
         context.input_format = input_format
         context.confidence_threshold = config.confidence_threshold
         config.confidence_threshold = 0.25
+        # param to be updated 
 
         if config.confidence_threshold is not None:
             processed_data = filter_by_confidence(data, config.confidence_threshold)