Update models/loaders/matanyone_loader.py

models/loaders/matanyone_loader.py

@@ -217,6 +217,7 @@ def _to_alpha(self, out_prob):
         if t.ndim == 3:
             return t[0] if t.shape[0] >= 1 else t.mean(0)
         return t
+
     def __call__(self, image, mask=None, **kwargs) -> np.ndarray:
         """
         Returns a 2-D float32 alpha [H,W]. On first call, provide a coarse mask.
@@ -306,7 +307,6 @@ def __exit__(self, *args): return False
         if msk_b1hw is not None:
             return _to_2d_alpha_numpy(msk_b1hw)
         return np.full((H, W), 0.5, dtype=np.float32)
-
 # -------------------------------- Loader ---------------------------------- #
 
 def _choose_precision(device: str) -> Tuple[torch.dtype, bool, Optional[torch.dtype]]:
@@ -431,6 +431,7 @@ def get_info(self) -> Dict[str, Any]:
 
     def debug_shapes(self, image, mask, tag: str = ""):
         debug_shapes(tag, image, mask)
+
 # -------------------------- Optional: Module-level symbols --------------------------
 
 __all__ = [
@@ -449,6 +450,7 @@ def debug_shapes(self, image, mask, tag: str = ""):
 
 if __name__ == "__main__":
     import sys
+    import cv2
 
     logging.basicConfig(level=logging.INFO)
     device = "cuda" if torch.cuda.is_available() else "cpu"