matrice 1.0.99146__py3-none-any.whl → 1.0.99148__py3-none-any.whl

matrice/deploy/utils/post_processing/__init__.py

@@ -119,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 (
@@ -170,6 +171,7 @@ from .usecases import (
     WindmillMaintenanceUseCase,
     CarPartSegmentationUseCase,
     FlowerUseCase,
+    SmokerDetectionUseCase,
 
     #Put all IMAGE based usecases here
     BloodCancerDetectionUseCase,
@@ -233,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()
@@ -297,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)
@@ -483,6 +487,7 @@ __all__ = [
     'LaneDetectionConfig',
     'WindmillMaintenanceConfig',
     'FlowerConfig',
+    'SmokerDetectionConfig',
 
     #Put all IMAGE based usecase CONFIGS here
     'BloodCancerDetectionConfig',
@@ -540,6 +545,7 @@ __all__ = [
     'LaneDetectionUseCase',
     'WindmillMaintenanceUseCase',
     'FlowerUseCase',
+    'SmokerDetectionUseCase',
 
     #Put all IMAGE based usecases here
     'BloodCancerDetectionUseCase',

matrice/deploy/utils/post_processing/config.py

@@ -29,6 +29,7 @@ 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 = {
@@ -62,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

@@ -448,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,
@@ -794,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."""
@@ -1618,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::
@@ -2003,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

@@ -83,6 +83,7 @@ from .usecases import (
     CarPartSegmentationUseCase,
     WindmillMaintenanceUseCase,
     FlowerUseCase,
+    SmokerDetectionUseCase,
 
     #Put all IMAGE based usecases here
     BloodCancerDetectionUseCase,
@@ -204,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)
@@ -354,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

@@ -66,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',
@@ -117,6 +118,7 @@ __all__ = [
     'LaneDetectionUseCase',
     'WindmillMaintenanceUseCase',
     'FlowerUseCase',
+    'SmokerDetectionUseCase',
 
     #Put all IMAGE based usecases here
     'BloodCancerDetectionUseCase',
@@ -173,6 +175,7 @@ __all__ = [
     'LaneDetectionConfig',
     'WindmillMaintenanceConfig',
     'FlowerConfig',
+    'SmokerDetectionConfig',