matrice-analytics 0.1.38__tar.gz → 0.1.39__tar.gz

{matrice_analytics-0.1.38 → matrice_analytics-0.1.39}/PKG-INFO

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

{matrice_analytics-0.1.38 → matrice_analytics-0.1.39}/matrice_analytics.egg-info/PKG-INFO

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

{matrice_analytics-0.1.38 → matrice_analytics-0.1.39}/src/matrice_analytics/post_processing/usecases/age_gender_detection.py

@@ -31,11 +31,94 @@ import numpy as np
 #import torch
 import re
 from matrice_analytics.post_processing.core.config import BaseConfig, AlertConfig, ZoneConfig
-try:
-    from PIL import Image
-    import onnxruntime as ort
-except:
-    print("Unable to import onnxruntime")
+# Auto-install and import required dependencies for age/gender detection
+def try_import_age_gender_deps():
+    """Try to import onnxruntime and PIL, auto-install if missing or fix NumPy issues."""
+    try:
+        import onnxruntime as ort
+        from PIL import Image
+        print("✓ Age/Gender dependencies imported successfully (onnxruntime, PIL)")
+        return ort, Image
+    except ImportError as e:
+        error_msg = str(e)
+        print(f"⚠ Import failed: {e}")
+
+        # Check if it's a NumPy version mismatch issue
+        if "NumPy" in error_msg and ("1.x" in error_msg or "2." in error_msg):
+            print("→ DETECTED: NumPy version mismatch issue!")
+            print("→ onnxruntime was compiled with NumPy 1.x but NumPy 2.x is installed")
+            print("→ Attempting to downgrade NumPy to version <2...")
+
+            import subprocess
+            import sys
+            try:
+                print("  Downgrading numpy...")
+                result = subprocess.run(
+                    [sys.executable, "-m", "pip", "install", "numpy<2", "--force-reinstall", "--no-warn-script-location"],
+                    capture_output=True,
+                    text=True,
+                    timeout=300
+                )
+                if result.returncode == 0:
+                    print("  ✓ NumPy downgraded successfully")
+                    print("→ Retrying onnxruntime import...")
+                    import onnxruntime as ort
+                    from PIL import Image
+                    print("✓ Dependencies imported after NumPy downgrade!")
+                    return ort, Image
+                else:
+                    print(f"  ✗ NumPy downgrade failed: {result.stderr}")
+                    return None, None
+            except Exception as numpy_fix_err:
+                print(f"✗ Failed to fix NumPy issue: {numpy_fix_err}")
+                return None, None
+
+        # Not a NumPy issue, try regular installation
+        print("→ Attempting auto-installation of missing packages...")
+        packages = []
+        try:
+            import onnxruntime
+        except ImportError:
+            packages.append("onnxruntime-gpu")
+
+        try:
+            from PIL import Image
+        except ImportError:
+            packages.append("pillow")
+
+        if packages:
+            print(f"→ Installing: {', '.join(packages)}")
+            import subprocess
+            import sys
+            try:
+                for pkg in packages:
+                    print(f"  Installing {pkg}...")
+                    result = subprocess.run(
+                        [sys.executable, "-m", "pip", "install", pkg, "--no-warn-script-location"],
+                        capture_output=True,
+                        text=True,
+                        timeout=300
+                    )
+                    if result.returncode == 0:
+                        print(f"  ✓ {pkg} installed")
+                    else:
+                        print(f"  ✗ Failed: {result.stderr}")
+
+                # Retry import
+                print("→ Retrying imports...")
+                import onnxruntime as ort
+                from PIL import Image
+                print("✓ Dependencies imported after auto-installation!")
+                return ort, Image
+            except Exception as install_err:
+                print(f"✗ Installation failed: {install_err}")
+                return None, None
+        return None, None
+    except Exception as e:
+        print(f"✗ Unexpected error: {e}")
+        return None, None
+
+ort, Image = try_import_age_gender_deps()
 
 
 def apply_category_mapping(results: Any, index_to_category: Dict[str, str]) -> Any:
@@ -89,9 +172,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 +243,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):