Update utils/cv_processing.py

utils/cv_processing.py

@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 """
 cv_processing.py · FIXED VERSION with proper SAM2 handling + MatAnyone stateful integration
+Now with environment variable support for USE_SAM2 and USE_MATANYONE
 
 All public functions in this module expect RGB images (H,W,3) unless stated otherwise.
 CoreVideoProcessor already converts BGR→RGB before calling into this module.
@@ -8,6 +9,7 @@
 
 from __future__ import annotations
 
+import os
 import logging
 from pathlib import Path
 from typing import Any, Dict, Optional, Tuple, Callable
@@ -17,6 +19,19 @@
 
 logger = logging.getLogger(__name__)
 
+# ----------------------------------------------------------------------------
+# Environment variable helpers
+# ----------------------------------------------------------------------------
+def _use_sam2_enabled() -> bool:
+    """Check if SAM2 should be used based on environment variable"""
+    val = os.getenv("USE_SAM2", "1")
+    return val.lower() in ("1", "true", "yes", "on")
+
+def _use_matanyone_enabled() -> bool:
+    """Check if MatAnyone should be used based on environment variable"""
+    val = os.getenv("USE_MATANYONE", "1")
+    return val.lower() in ("1", "true", "yes", "on")
+
 # ----------------------------------------------------------------------------
 # Background presets
 # ----------------------------------------------------------------------------
@@ -187,10 +202,15 @@ def segment_person_hq(
     High-quality person segmentation with proper SAM2 handling.
     Expects RGB frame (H,W,3), uint8 or float in [0,1].
     """
+    # Override with environment variable if not explicitly set
+    if use_sam2 is None:
+        use_sam2 = _use_sam2_enabled()
+    
     frame_rgb = _ensure_rgb(frame)
     h, w = frame_rgb.shape[:2]
 
     if use_sam2 is False:
+        logger.info("SAM2 disabled by environment variable, using fallback segmentation")
         return _simple_person_segmentation(frame_rgb)
 
     if predictor is not None:
@@ -251,6 +271,8 @@ def segment_person_hq(
 
                     if mask is not None:
                         mask = _to_mask01(mask)
+                        # Add debug logging
+                        logger.info(f"SAM2 mask stats: shape={mask.shape}, min={mask.min():.3f}, max={mask.max():.3f}, mean={mask.mean():.3f}")
                         if float(mask.max()) > 0.1:
                             return np.ascontiguousarray(mask)
                         else:
@@ -294,9 +316,14 @@ def refine_mask_hq(
     Returns:
       2-D float32 alpha [H,W], contiguous, in [0,1] (OpenCV-safe).
     """
+    # Override with environment variable if not explicitly set
+    if use_matanyone is None:
+        use_matanyone = _use_matanyone_enabled()
+    
     mask01 = _to_mask01(mask)
 
     if use_matanyone is False:
+        logger.info("MatAnyone disabled by environment variable, returning unrefined mask")
         return mask01
 
     if matanyone is not None and callable(matanyone):
@@ -470,4 +497,4 @@ def validate_video_file(video_path: str) -> Tuple[bool, str]:
     "create_professional_background",
     "validate_video_file",
     "PROFESSIONAL_BACKGROUNDS",
-]
+]