matrice 1.0.99144__tar.gz → 1.0.99146__tar.gz

{matrice-1.0.99144/src/matrice.egg-info → matrice-1.0.99146}/PKG-INFO

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

{matrice-1.0.99144 → matrice-1.0.99146}/setup.py

@@ -14,7 +14,7 @@ long_description = (here / "README.md").read_text(encoding="utf-8")
 setup(
     name="matrice",
 
-    version = "1.0.99144",
+    version = "1.0.99146",
     
     description="SDK for connecting to matrice.ai services",
     long_description=long_description,

{matrice-1.0.99144 → matrice-1.0.99146}/src/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
@@ -193,6 +195,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()
@@ -256,6 +259,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)

{matrice-1.0.99144 → matrice-1.0.99146}/src/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",
@@ -33,6 +34,7 @@ APP_NAME_TO_USECASE = {
 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",

{matrice-1.0.99144 → matrice-1.0.99146}/src/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,
@@ -592,6 +593,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 +602,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:
@@ -972,6 +982,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
@@ -1759,6 +1782,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

{matrice-1.0.99144 → matrice-1.0.99146}/src/matrice/deploy/utils/post_processing/processor.py

@@ -51,6 +51,7 @@ from .usecases import (
     ParkingUseCase,
     FaceEmotionUseCase,
     UnderwaterPlasticUseCase,
+    PipelineDetectionUseCase,
     PedestrianDetectionUseCase,
     ChickenPoseDetectionUseCase,
     TheftDetectionUseCase,
@@ -175,6 +176,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)

{matrice-1.0.99144 → matrice-1.0.99146}/src/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