Refactor mask loading method in ImageProcessor

soundimage/utils/image_processor.py

@@ -28,7 +28,12 @@ https://stackoverflow.com/questions/23853632/which-kind-of-interpolation-best-fo
 """
 
 
-
+def load_fixed_mask(resolution: int, mask_path: str) -> torch.Tensor:
+    mask_image = cv2.imread(mask_path)
+    mask_image = cv2.cvtColor(mask_image, cv2.COLOR_BGR2RGB)
+    mask_image = cv2.resize(mask_image, (resolution, resolution), interpolation=cv2.INTER_AREA) / 255.0
+    mask_image = rearrange(torch.from_numpy(mask_image), "h w c -> c h w")
+    return mask_image
 
 
 class ImageProcessor:
@@ -61,12 +66,8 @@ class ImageProcessor:
                 # self.face_mesh = mp.solutions.face_mesh.FaceMesh(static_image_mode=True)  # Process single image
                 self.face_mesh = None
                 self.fa = None
-    def load_fixed_mask(resolution: int, mask_path: str) -> torch.Tensor:
-        mask_image = cv2.imread(mask_path)
-        mask_image = cv2.cvtColor(mask_image, cv2.COLOR_BGR2RGB)
-        mask_image = cv2.resize(mask_image, (resolution, resolution), interpolation=cv2.INTER_AREA) / 255.0
-        mask_image = rearrange(torch.from_numpy(mask_image), "h w c -> c h w")
-        return mask_image
+
+
 
     def detect_facial_landmarks(self, image: np.ndarray):
         height, width, _ = image.shape