matrice 1.0.99145__py3-none-any.whl → 1.0.99147__py3-none-any.whl

matrice/deploy/utils/post_processing/__init__.py

@@ -71,11 +71,13 @@ from .usecases.license_plate_detection import LicensePlateConfig
 from .usecases.pothole_segmentation import PotholeConfig
 from .usecases.wound_segmentation import WoundConfig, WoundSegmentationUseCase
 from .usecases.face_emotion import FaceEmotionConfig
+from .usecases.pipeline_detection import PipelineDetectionUseCase
 from .usecases.parking_space_detection import ParkingSpaceConfig
 from .usecases.underwater_pollution_detection import UnderwaterPlasticConfig
 from .usecases.pedestrian_detection import PedestrianDetectionConfig
 from .usecases.age_detection import AgeDetectionConfig
 from .usecases.mask_detection import MaskDetectionConfig
+from .usecases.pipeline_detection import PipelineDetectionConfig
 from .usecases.chicken_pose_detection import ChickenPoseDetectionConfig
 from .usecases.field_mapping import FieldMappingConfig, FieldMappingUseCase
 from .usecases.leaf_disease import LeafDiseaseDetectionConfig, LeafDiseaseDetectionUseCase
@@ -117,6 +119,7 @@ from .usecases.leaf import LeafConfig, LeafUseCase
 from .usecases.blood_cancer_detection_img import BloodCancerDetectionConfig, BloodCancerDetectionUseCase
 from .usecases.skin_cancer_classification_img import SkinCancerClassificationConfig, SkinCancerClassificationUseCase
 from .usecases.plaque_segmentation_img import PlaqueSegmentationConfig, PlaqueSegmentationUseCase
+from .usecases.smoker_detection import SmokerDetectionUseCase, SmokerDetectionConfig
 
 # Use case implementations
 from .usecases import (
@@ -168,6 +171,7 @@ from .usecases import (
     WindmillMaintenanceUseCase,
     CarPartSegmentationUseCase,
     FlowerUseCase,
+    SmokerDetectionUseCase,
 
     #Put all IMAGE based usecases here
     BloodCancerDetectionUseCase,
@@ -193,6 +197,7 @@ _underwater_pollution_detection = UnderwaterPlasticUseCase()
 _pedestrian_detection = PedestrianDetectionUseCase()
 _age_detection = AgeDetectionUseCase()
 _mask_detection = MaskDetectionUseCase()
+_pipeline_detection = PipelineDetectionUseCase()
 _banana_defect_detection = BananaMonitoringUseCase()
 _chicken_pose_detection = ChickenPoseDetectionUseCase()
 _theft_detection = TheftDetectionUseCase()
@@ -230,6 +235,7 @@ _wound_segmentation = WoundSegmentationUseCase()
 _leaf_disease = LeafDiseaseDetectionUseCase()
 _flower_segmentation = FlowerUseCase()
 _leaf_det = LeafUseCase()
+_smoker_detection = SmokerDetectionUseCase()
 
 #Put all IMAGE based usecases here
 _blood_cancer_detection = BloodCancerDetectionUseCase()
@@ -256,6 +262,7 @@ registry.register_use_case(_age_detection.category, _age_detection.name, AgeDete
 registry.register_use_case(_pricetag_detection.category, _pricetag_detection.name, PriceTagUseCase)
 registry.register_use_case(_weld_defect_detection.category, _weld_defect_detection.name, WeldDefectUseCase  )
 registry.register_use_case(_mask_detection.category, _mask_detection.name, MaskDetectionUseCase)
+registry.register_use_case(_pipeline_detection.category, _pipeline_detection.name, PipelineDetectionUseCase)
 registry.register_use_case(_banana_defect_detection.category, _banana_defect_detection.name, BananaMonitoringUseCase)
 registry.register_use_case(_chicken_pose_detection.category, _chicken_pose_detection.name, ChickenPoseDetectionUseCase)
 registry.register_use_case(_theft_detection.category, _theft_detection.name, TheftDetectionUseCase)
@@ -293,6 +300,7 @@ registry.register_use_case(_leaf_disease.category, _leaf_disease.name, LeafDisea
 registry.register_use_case(_flower_segmentation.category, _flower_segmentation.name, FlowerUseCase)
 registry.register_use_case(_parking_det.category, _parking_det.name, ParkingUseCase)
 registry.register_use_case(_leaf_det.category, _leaf_det.name, LeafUseCase)
+registry.register_use_case(_smoker_detection.category, _smoker_detection.name, SmokerDetectionUseCase)
 
 #Put all IMAGE based usecases here
 registry.register_use_case(_blood_cancer_detection.category, _blood_cancer_detection.name, BloodCancerDetectionUseCase)
@@ -479,6 +487,7 @@ __all__ = [
     'LaneDetectionConfig',
     'WindmillMaintenanceConfig',
     'FlowerConfig',
+    'SmokerDetectionConfig',
 
     #Put all IMAGE based usecase CONFIGS here
     'BloodCancerDetectionConfig',
@@ -536,6 +545,7 @@ __all__ = [
     'LaneDetectionUseCase',
     'WindmillMaintenanceUseCase',
     'FlowerUseCase',
+    'SmokerDetectionUseCase',
 
     #Put all IMAGE based usecases here
     'BloodCancerDetectionUseCase',

matrice/deploy/utils/post_processing/config.py

@@ -1,6 +1,7 @@
 APP_NAME_TO_USECASE = {
     "people_counting": "people_counting",
     "mask_detection": "mask_detection",
+    "pipeline_detection": "pipeline_detection",
     "vehicle_monitoring": "vehicle_monitoring",
     "vehicle_type_monitoring": "vehicle_monitoring",
     "weapon_detection": "weapon_detection",
@@ -28,11 +29,13 @@ APP_NAME_TO_USECASE = {
     "concrete_crack_detection": "concrete_crack_detection",
     "lane_detection" : "lane_detection",
     "shelf_inventory" :"shelf_inventory",
+    "smoker_detection": "smoker_detection"
 }
 
 APP_NAME_TO_CATEGORY = {
     "people_counting": "general",
     "mask_detection": "mask_detection",
+    "pipeline_detection": "pipeline_detection",
     "vehicle_monitoring": "traffic",
     "vehicle_type_monitoring": "traffic",
     "weapon_detection": "security",
@@ -60,6 +63,7 @@ APP_NAME_TO_CATEGORY = {
     "concrete_crack_detection": "general",
     "lane_detection" : "traffic",
     "shelf_inventory" : "retail",
+    "smoker_detection": "general"
 }
 
 def get_usecase_from_app_name(app_name: str) -> str:

matrice/deploy/utils/post_processing/core/config.py

@@ -415,6 +415,7 @@ class ConfigManager:
             "fire_smoke_detection": None,
             "flare_analysis" : None,
             "mask_detection": None,
+            "pipeline_detection": None,
             "parking_space_detection": None,
             "car_damage_detection":None,
             "weld_defect_detection" : None,
@@ -447,6 +448,7 @@ class ConfigManager:
             'windmill_maintenance': None,
             'face_emotion': None,
             'flower_segmentation': None,
+            'smoker_detection': None,
 
             #Put all image based usecases here::
             'blood_cancer_detection_img': None,
@@ -592,6 +594,7 @@ class ConfigManager:
             return ParkingSpaceConfig
         except ImportError:
             return None
+
     def _get_mask_detection_config_class(self):
         """Register a configuration class for a use case."""
         try:
@@ -600,6 +603,14 @@ class ConfigManager:
         except ImportError:
             return None
 
+    def _get_pipeline_detection_config_class(self):
+        """Register a configuration class for a use case."""
+        try:
+            from ..usecases.pipeline_detection import PipelineDetectionConfig
+            return PipelineDetectionConfig
+        except ImportError:
+            return None
+
     def _get_pothole_segmentation_config_class(self):
         """Register a configuration class for a use case."""
         try:
@@ -784,6 +795,14 @@ class ConfigManager:
         except ImportError:
             return None
     
+    def smoker_detection_config_class(self):
+        """Register a configuration class for a use case."""
+        try:
+            from ..usecases.smoker_detection import SmokerDetectionConfig
+            return SmokerDetectionConfig
+        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."""
@@ -972,6 +991,19 @@ class ConfigManager:
                 alert_config=alert_config,
                 **kwargs
             )
+        elif usecase == "pipeline_detection":
+            from ..usecases.pipeline_detection import PipelineDetectionConfig
+
+            alert_config = kwargs.pop("alert_config", None)
+            if alert_config and isinstance(alert_config, dict):
+                alert_config = AlertConfig(**alert_config)
+
+            config = PipelineDetectionConfig(
+                category=category or "pipeline_detection",
+                usecase=usecase,
+                alert_config=alert_config,
+                **kwargs
+            )
         elif usecase == "shoplifting_detection":
             # Import here to avoid circular import
             from ..usecases.shoplifting_detection import ShopliftingDetectionConfig
@@ -1595,6 +1627,22 @@ class ConfigManager:
                 alert_config=alert_config,
                 **kwargs
             )
+        
+        elif usecase == "smoker_detection":
+            # Import here to avoid circular import
+            from ..usecases.smoker_detection import SmokerDetectionConfig
+
+            # Handle nested configurations
+            alert_config = kwargs.pop("alert_config", None)
+            if alert_config and isinstance(alert_config, dict):
+                alert_config = AlertConfig(**alert_config)
+
+            config = SmokerDetectionConfig(
+                category=category or "general",
+                usecase=usecase,
+                alert_config=alert_config,
+                **kwargs
+            )
 
         
         #Add IMAGE based usecases here::
@@ -1759,6 +1807,11 @@ class ConfigManager:
             from ..usecases.mask_detection import MaskDetectionConfig
             default_config = MaskDetectionConfig()
             return default_config.to_dict()
+        
+        elif usecase == "pipeline_detection":
+            from ..usecases.pipeline_detection import PipelineDetectionConfig
+            default_config = PipelineDetectionConfig()
+            return default_config.to_dict()
 
         elif usecase == "fire_smoke_detection":
             # Import here to avoid circular import
@@ -1975,6 +2028,11 @@ class ConfigManager:
             from ..usecases.flower_segmentation import FlowerConfig
             default_config = FlowerConfig()
             return default_config.to_dict()
+        elif usecase == "smoker_detection":
+            # Import here to avoid circular import
+            from ..usecases.smoker_detection import SmokerDetectionConfig
+            default_config = SmokerDetectionConfig()
+            return default_config.to_dict()
 
         #Add all image based usecases here
         elif usecase == "blood_cancer_detection_img":

matrice/deploy/utils/post_processing/processor.py

@@ -51,6 +51,7 @@ from .usecases import (
     ParkingUseCase,
     FaceEmotionUseCase,
     UnderwaterPlasticUseCase,
+    PipelineDetectionUseCase,
     PedestrianDetectionUseCase,
     ChickenPoseDetectionUseCase,
     TheftDetectionUseCase,
@@ -82,6 +83,7 @@ from .usecases import (
     CarPartSegmentationUseCase,
     WindmillMaintenanceUseCase,
     FlowerUseCase,
+    SmokerDetectionUseCase,
 
     #Put all IMAGE based usecases here
     BloodCancerDetectionUseCase,
@@ -175,6 +177,7 @@ class PostProcessor:
         registry.register_use_case("weld", "weld_defect_detection", WeldDefectUseCase)
         registry.register_use_case("price_tag", "price_tag_detection", PriceTagUseCase)
         registry.register_use_case("mask_detection", "mask_detection", MaskDetectionUseCase)
+        registry.register_use_case("pipeline_detection", "pipeline_detection", PipelineDetectionUseCase)
         registry.register_use_case("automobile", "distracted_driver_detection", DistractedDriverUseCase)
         registry.register_use_case("traffic", "emergency_vehicle_detection", EmergencyVehicleUseCase)
         registry.register_use_case("energy", "solar_panel", SolarPanelUseCase)
@@ -202,6 +205,7 @@ class PostProcessor:
         registry.register_use_case("agriculture", "flower_segmentation", FlowerUseCase)
         registry.register_use_case("general", "parking_det", ParkingUseCase)
         registry.register_use_case("agriculture", "leaf_det", LeafUseCase)
+        registry.register_use_case("general", "smoker_detection", SmokerDetectionUseCase)
 
         #Put all IMAGE based usecases here
         registry.register_use_case("healthcare", "bloodcancer_img_detection", BloodCancerDetectionUseCase)
@@ -352,8 +356,8 @@ class PostProcessor:
                 result = use_case.process(data, parsed_config, context, stream_info)
             elif isinstance(use_case, LeafUseCase):
                 result = use_case.process(data, parsed_config, context, stream_info)
-
-            
+            elif isinstance(use_case, SmokerDetectionUseCase):
+                result = use_case.process(data, parsed_config, context, stream_info)
 
             
             #Put all IMAGE based usecases here

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

@@ -28,6 +28,7 @@ from .banana_defect_detection import BananaMonitoringUseCase,BananaMonitoringCon
 from .car_damage_detection import CarDamageConfig, CarDamageDetectionUseCase
 from .price_tag_detection import PriceTagConfig, PriceTagUseCase
 from .mask_detection import MaskDetectionConfig, MaskDetectionUseCase
+from .pipeline_detection import PipelineDetectionConfig, PipelineDetectionUseCase
 from .distracted_driver_detection import DistractedDriverUseCase, DistractedDriverConfig
 from .emergency_vehicle_detection import EmergencyVehicleUseCase, EmergencyVehicleConfig
 from .solar_panel import SolarPanelUseCase, SolarPanelConfig
@@ -65,6 +66,7 @@ from .leaf import LeafConfig, LeafUseCase
 from .blood_cancer_detection_img import BloodCancerDetectionConfig, BloodCancerDetectionUseCase
 from .skin_cancer_classification_img import SkinCancerClassificationConfig, SkinCancerClassificationUseCase
 from .plaque_segmentation_img import PlaqueSegmentationConfig, PlaqueSegmentationUseCase
+from .smoker_detection import SmokerDetectionConfig, SmokerDetectionUseCase
 
 __all__ = [
     'PeopleCountingUseCase',
@@ -116,6 +118,7 @@ __all__ = [
     'LaneDetectionUseCase',
     'WindmillMaintenanceUseCase',
     'FlowerUseCase',
+    'SmokerDetectionUseCase',
 
     #Put all IMAGE based usecases here
     'BloodCancerDetectionUseCase',
@@ -172,6 +175,7 @@ __all__ = [
     'LaneDetectionConfig',
     'WindmillMaintenanceConfig',
     'FlowerConfig',
+    'SmokerDetectionConfig',