matrice-analytics 0.1.38__py3-none-any.whl → 0.1.40__py3-none-any.whl

matrice_analytics/post_processing/usecases/age_gender_detection.py

@@ -31,11 +31,31 @@ import numpy as np
 #import torch
 import re
 from matrice_analytics.post_processing.core.config import BaseConfig, AlertConfig, ZoneConfig
+
+# Try to import optional age/gender dependencies (fail gracefully if missing)
+ort = None
+Image = None
+
 try:
-    from PIL import Image
     import onnxruntime as ort
-except:
-    print("Unable to import onnxruntime")
+    from PIL import Image
+    print("✓ Age/Gender dependencies available (onnxruntime, PIL)")
+except ImportError as e:
+    error_msg = str(e)
+    print(f"⚠ Age/Gender dependencies not available: {e}")
+
+    # Check if it's a NumPy version mismatch issue and provide helpful message
+    if "NumPy" in error_msg and ("1.x" in error_msg or "2." in error_msg):
+        print("→ DETECTED: NumPy version mismatch!")
+        print("  onnxruntime was compiled with NumPy 1.x but NumPy 2.x is installed")
+        print("  To fix manually: pip install 'numpy<2' --force-reinstall")
+    else:
+        print("  To enable manually: pip install onnxruntime-gpu pillow")
+
+    print("→ Age/Gender detection will be disabled")
+except Exception as e:
+    print(f"⚠ Error importing Age/Gender dependencies: {e}")
+    print("→ Age/Gender detection will be disabled")
 
 
 def apply_category_mapping(results: Any, index_to_category: Dict[str, str]) -> Any:
@@ -89,9 +109,16 @@ def load_model_from_checkpoint(checkpoint_path,local_path):
     """
     Load a model from checkpoint URL
     """
+    if ort is None:
+        raise RuntimeError(
+            "onnxruntime is not available. Cannot load age/gender detection models.\n"
+            "Please install: pip install onnxruntime-gpu\n"
+            "Or downgrade numpy: pip install 'numpy<2'"
+        )
+
     try:
         print(f"Loading model from checkpoint: {checkpoint_path}")
-        
+
         # Check if checkpoint is a URL
         if checkpoint_path.startswith(('http://', 'https://')):
             # Download checkpoint from URL
@@ -153,10 +180,25 @@ class AgeGenderConfig(BaseConfig):
 
     def __post_init__(self):
         if self.gender_url and self.age_url:
-            self.age_url = load_model_from_checkpoint(self.age_url, "age_detection_prod_model.onnx")
-            self.gender_url = load_model_from_checkpoint(self.gender_url, "gender_detection_prod_model.onnx")
+            try:
+                self.age_url = load_model_from_checkpoint(self.age_url, "age_detection_prod_model.onnx")
+                self.gender_url = load_model_from_checkpoint(self.gender_url, "gender_detection_prod_model.onnx")
+                print("✓ Age and Gender models loaded successfully")
+            except RuntimeError as e:
+                print(f"✗ Failed to load Age/Gender models: {e}")
+                print("→ Age/Gender detection will be disabled")
+                self.age_url = None
+                self.gender_url = None
+            except Exception as e:
+                print(f"✗ Unexpected error loading Age/Gender models: {e}")
+                import traceback
+                traceback.print_exc()
+                self.age_url = None
+                self.gender_url = None
         else:
-            raise ValueError("Age and Gender model URL must be passed!")
+            print("⚠ Age and Gender model URLs not provided - detection will be disabled")
+            self.age_url = None
+            self.gender_url = None
         
 
 class AgeGenderUseCase(BaseProcessor):

matrice_analytics/post_processing/usecases/color/clip.py

@@ -29,89 +29,103 @@ import cv2
 import io
 import threading
 
-# Auto-install and import required dependencies for CLIP color detection
-def try_import_with_auto_install():
+# Try to import optional CLIP dependencies (fail gracefully if missing)
+# These will be None if not available, and auto-install will be attempted on first use
+ort = None
+Image = None
+CLIPProcessor = None
+ir_files = None
+ir_as_file = None
+
+try:
+    import onnxruntime as ort
+    from PIL import Image
+    from transformers import CLIPProcessor
+    from importlib.resources import files as ir_files, as_file as ir_as_file
+    print("✓ CLIP dependencies available (onnxruntime, PIL, transformers)")
+except ImportError as e:
+    print(f"⚠ CLIP dependencies not available at import time: {e}")
+    print("→ Will attempt auto-installation when color detection is first used")
+except Exception as e:
+    print(f"⚠ Error importing CLIP dependencies: {e}")
+    print("→ Color detection may be disabled")
+
+def try_install_clip_dependencies():
     """
-    Try to import required dependencies, auto-install if missing, then try again.
-    Returns tuple of (ort, Image, CLIPProcessor, ir_files, ir_as_file)
+    Attempt to install missing CLIP dependencies.
+    Only called when ClipProcessor is actually instantiated (lazy installation).
+    Returns True if successful, False otherwise.
     """
-    # First attempt: try to import
+    global ort, Image, CLIPProcessor, ir_files, ir_as_file
+
+    print("→ Color detection is being used but dependencies are missing")
+    print("→ Attempting auto-installation of missing packages...")
+
+    packages_to_install = []
+
+    # Check which packages are missing
     try:
-        import onnxruntime as ort
-        from PIL import Image
-        from transformers import CLIPProcessor
-        from importlib.resources import files as ir_files, as_file as ir_as_file
-        print("✓ CLIP dependencies imported successfully (onnxruntime, PIL, transformers)")
-        return ort, Image, CLIPProcessor, ir_files, ir_as_file
-    except ImportError as e:
-        print(f"⚠ Import failed on first attempt: {e}")
-        print("→ Attempting auto-installation of missing packages...")
+        import onnxruntime
+    except ImportError:
+        packages_to_install.append("onnxruntime-gpu")
 
-        # Try to install missing packages
-        packages_to_install = []
+    try:
+        from PIL import Image as PILImage
+    except ImportError:
+        packages_to_install.append("pillow")
 
-        # Check which specific packages are missing
-        try:
-            import onnxruntime
-        except ImportError:
-            packages_to_install.append("onnxruntime-gpu")
+    try:
+        from transformers import CLIPProcessor as CLIP
+    except ImportError:
+        packages_to_install.append("transformers")
 
-        try:
-            from PIL import Image
-        except ImportError:
-            packages_to_install.append("pillow")
+    if not packages_to_install:
+        print("→ All packages are available, retrying import...")
+    else:
+        print(f"→ Installing: {', '.join(packages_to_install)}")
 
+        import subprocess
         try:
-            from transformers import CLIPProcessor
-        except ImportError:
-            packages_to_install.append("transformers")
+            for package in packages_to_install:
+                print(f"  Installing {package}...")
+                result = subprocess.run(
+                    [sys.executable, "-m", "pip", "install", package, "--no-warn-script-location"],
+                    capture_output=True,
+                    text=True,
+                    timeout=300
+                )
+                if result.returncode == 0:
+                    print(f"  ✓ {package} installed successfully")
+                else:
+                    print(f"  ✗ Failed to install {package}: {result.stderr}")
+                    return False
+        except Exception as install_error:
+            print(f"✗ Installation failed: {install_error}")
+            return False
 
-        if packages_to_install:
-            print(f"→ Installing: {', '.join(packages_to_install)}")
-            try:
-                # Use sys.executable to ensure we install to the correct Python environment
-                for package in packages_to_install:
-                    print(f"  Installing {package}...")
-                    result = subprocess.run(
-                        [sys.executable, "-m", "pip", "install", package, "--no-warn-script-location"],
-                        capture_output=True,
-                        text=True,
-                        timeout=300  # 5 minute timeout
-                    )
-                    if result.returncode == 0:
-                        print(f"  ✓ {package} installed successfully")
-                    else:
-                        print(f"  ✗ Failed to install {package}: {result.stderr}")
-
-                # Second attempt: try to import after installation
-                print("→ Retrying imports after installation...")
-                try:
-                    import onnxruntime as ort
-                    from PIL import Image
-                    from transformers import CLIPProcessor
-                    from importlib.resources import files as ir_files, as_file as ir_as_file
-                    print("✓ CLIP dependencies imported successfully after auto-installation!")
-                    return ort, Image, CLIPProcessor, ir_files, ir_as_file
-                except ImportError as e2:
-                    print(f"✗ Import still failed after installation: {e2}")
-                    print("→ Color detection will be disabled")
-                    return None, None, None, None, None
-
-            except subprocess.TimeoutExpired:
-                print("✗ Installation timeout - packages took too long to install")
-                return None, None, None, None, None
-            except Exception as install_error:
-                print(f"✗ Installation error: {install_error}")
-                return None, None, None, None, None
-        else:
-            print("✗ Unable to determine which packages are missing")
-            return None, None, None, None, None
+    # Retry imports after installation
+    try:
+        print("→ Retrying imports after installation...")
+        import onnxruntime as ort_module
+        from PIL import Image as PILImage
+        from transformers import CLIPProcessor as CLIPProc
+        from importlib.resources import files as ir_files_module, as_file as ir_as_file_module
+
+        # Update global variables
+        ort = ort_module
+        Image = PILImage
+        CLIPProcessor = CLIPProc
+        ir_files = ir_files_module
+        ir_as_file = ir_as_file_module
+
+        print("✓ CLIP dependencies imported successfully after installation!")
+        return True
+    except ImportError as e:
+        print(f"✗ Import still failed after installation: {e}")
+        return False
     except Exception as e:
-        print(f"✗ Unexpected error during import: {e}")
-        return None, None, None, None, None
-
-# Execute the auto-install import
-ort, Image, CLIPProcessor, ir_files, ir_as_file = try_import_with_auto_install()
+        print(f"✗ Unexpected error: {e}")
+        return False
 
 def load_model_from_checkpoint(checkpoint_url: str, providers: Optional[List] = None):
     """
@@ -186,16 +200,22 @@ class ClipProcessor:
                  processor_dir: Optional[str] = None,
                  providers: Optional[List[str]] = None):
 
-        # Check if required dependencies are available
+        # Check if required dependencies are available, try auto-install if not
         if ort is None or CLIPProcessor is None or Image is None:
-            raise RuntimeError(
-                "Required dependencies for ClipProcessor are not available.\n"
-                "Missing: " +
-                (("onnxruntime " if ort is None else "") +
-                 ("transformers(CLIPProcessor) " if CLIPProcessor is None else "") +
-                 ("PIL(Image) " if Image is None else "")).strip() + "\n"
-                "Please install: pip install transformers onnxruntime-gpu pillow"
-            )
+            print("⚠ Color detection dependencies missing, attempting auto-installation...")
+
+            # Try to auto-install missing dependencies (lazy installation)
+            if not try_install_clip_dependencies():
+                raise RuntimeError(
+                    "Required dependencies for ClipProcessor are not available.\n"
+                    "Auto-installation failed. Missing: " +
+                    (("onnxruntime " if ort is None else "") +
+                     ("transformers(CLIPProcessor) " if CLIPProcessor is None else "") +
+                     ("PIL(Image) " if Image is None else "")).strip() + "\n"
+                    "Please install manually: pip install transformers onnxruntime-gpu pillow"
+                )
+
+            print("✓ Auto-installation successful, continuing with ClipProcessor initialization")
 
         self.color_category: List[str] = ["black", "white", "yellow", "gray", "red", "blue", "light blue",
         "green", "brown"]

{matrice_analytics-0.1.38.dist-info → matrice_analytics-0.1.40.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: matrice_analytics
-Version: 0.1.38
+Version: 0.1.40
 Summary: Common server utilities for Matrice.ai services
 Author-email: "Matrice.ai" <dipendra@matrice.ai>
 License-Expression: MIT

{matrice_analytics-0.1.38.dist-info → matrice_analytics-0.1.40.dist-info}/RECORD

@@ -87,7 +87,7 @@ matrice_analytics/post_processing/usecases/__init__.py,sha256=tsu51e2q2gbr5UcALe
 matrice_analytics/post_processing/usecases/abandoned_object_detection.py,sha256=zVrqlvgsjc5F2JHd-Zav6d-poVIpTJBpPKZ3CC71lQk,40332
 matrice_analytics/post_processing/usecases/advanced_customer_service.py,sha256=ELt5euxr6P4X2s8-YGngmj27QscOHefjOsx3774sNFk,75914
 matrice_analytics/post_processing/usecases/age_detection.py,sha256=yn1LXOgbnOWSMDnsCds6-uN6W-I1Hy4_-AMrjbT5PtY,41318
-matrice_analytics/post_processing/usecases/age_gender_detection.py,sha256=JSmQOSXdERZ6lOa28OrYBMEWa6Ey6VGv_mLjrRHIgts,50703
+matrice_analytics/post_processing/usecases/age_gender_detection.py,sha256=ok8uCG8pHKqouTSOQyUANou29k6OOohxnhFsTwayOL0,52533
 matrice_analytics/post_processing/usecases/anti_spoofing_detection.py,sha256=XdtDdXGzZMLQdWcoOoiE5t4LPYHhgOtJ7tZCNlq1A2E,31329
 matrice_analytics/post_processing/usecases/assembly_line_detection.py,sha256=I_oeuec84KJnGMg_A8Wgs9U6h_IiopkDz9FbM1JG110,40410
 matrice_analytics/post_processing/usecases/banana_defect_detection.py,sha256=uKvB550Xve3iWfdWeOU3VhgORxD-M_UzsQDQKkYcFBc,30338
@@ -166,7 +166,7 @@ matrice_analytics/post_processing/usecases/weld_defect_detection.py,sha256=b0dAJ
 matrice_analytics/post_processing/usecases/wildlife_monitoring.py,sha256=TMVHJ5GLezmqG7DywmqbLggqNXgpsb63MD7IR6kvDkk,43446
 matrice_analytics/post_processing/usecases/windmill_maintenance.py,sha256=G1eqo3Z-HYmGJ6oeZYrpZwhpvqQ9Lc_T-6S7BLBXHeA,40498
 matrice_analytics/post_processing/usecases/wound_segmentation.py,sha256=ehNX6VuWMB3xAnCySO3ra3Tf_5FUNg5LCSdq_91h374,38342
-matrice_analytics/post_processing/usecases/color/clip.py,sha256=ZUqj0s0KFcU6kZ21hpiQWO3CRqn8s5EZ3IhWTskHbS8,21322
+matrice_analytics/post_processing/usecases/color/clip.py,sha256=12ir0RkJCb4FJErn_e_9UbjtbBki5PMH2hWJLzyV75Q,21471
 matrice_analytics/post_processing/usecases/color/color_map_utils.py,sha256=SP-AEVcjLmL8rxblu-ixqUJC2fqlcr7ab4hWo4Fcr_k,2677
 matrice_analytics/post_processing/usecases/color/color_mapper.py,sha256=nKPc28mJLlrl2HPua5EUUMweRRSI6WrrUBkeitTm7ms,17459
 matrice_analytics/post_processing/usecases/color/clip_processor/merges.txt,sha256=n9aR98gDkhDg_O0VhlRmxlgg0JtjmIsBdL_iXeKZBRo,524619
@@ -188,8 +188,8 @@ matrice_analytics/post_processing/utils/format_utils.py,sha256=UTF7A5h9j0_S12xH9
 matrice_analytics/post_processing/utils/geometry_utils.py,sha256=BWfdM6RsdJTTLR1GqkWfdwpjMEjTCJyuBxA4zVGKdfk,9623
 matrice_analytics/post_processing/utils/smoothing_utils.py,sha256=78U-yucAcjUiZ0NIAc9NOUSIT0PWP1cqyIPA_Fdrjp0,14699
 matrice_analytics/post_processing/utils/tracking_utils.py,sha256=rWxuotnJ3VLMHIBOud2KLcu4yZfDp7hVPWUtNAq_2xw,8288
-matrice_analytics-0.1.38.dist-info/licenses/LICENSE.txt,sha256=_uQUZpgO0mRYL5-fPoEvLSbNnLPv6OmbeEDCHXhK6Qc,1066
-matrice_analytics-0.1.38.dist-info/METADATA,sha256=486XlgiBZP14Sdp7HlHG1T4B-vPjNYRhYvMC4m0JPNs,14378
-matrice_analytics-0.1.38.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-matrice_analytics-0.1.38.dist-info/top_level.txt,sha256=STAPEU-e-rWTerXaspdi76T_eVRSrEfFpURSP7_Dt8E,18
-matrice_analytics-0.1.38.dist-info/RECORD,,
+matrice_analytics-0.1.40.dist-info/licenses/LICENSE.txt,sha256=_uQUZpgO0mRYL5-fPoEvLSbNnLPv6OmbeEDCHXhK6Qc,1066
+matrice_analytics-0.1.40.dist-info/METADATA,sha256=tROotN2KHqbTIpf18BknZQ8Z4zO8VHNMcf7f1q4svas,14378
+matrice_analytics-0.1.40.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+matrice_analytics-0.1.40.dist-info/top_level.txt,sha256=STAPEU-e-rWTerXaspdi76T_eVRSrEfFpURSP7_Dt8E,18
+matrice_analytics-0.1.40.dist-info/RECORD,,