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

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):

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

@@ -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.dist-info → matrice_analytics-0.1.39.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=liWcwP4-rC97iVH2wzeDTdCPud-eZ5ZKg4odHLIrI4M,55131
 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
@@ -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.39.dist-info/licenses/LICENSE.txt,sha256=_uQUZpgO0mRYL5-fPoEvLSbNnLPv6OmbeEDCHXhK6Qc,1066
+matrice_analytics-0.1.39.dist-info/METADATA,sha256=3btcsrjFcTzAkncEjVaPi4hUNRHDBFe34x4uJ6Mf1Kg,14378
+matrice_analytics-0.1.39.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+matrice_analytics-0.1.39.dist-info/top_level.txt,sha256=STAPEU-e-rWTerXaspdi76T_eVRSrEfFpURSP7_Dt8E,18
+matrice_analytics-0.1.39.dist-info/RECORD,,