changes

soundimage/utils/image_processor.py

@@ -65,11 +65,11 @@ class ImageProcessor:
                 
     def load_fixed_mask(resolution: int, mask_path: str) -> torch.Tensor:
         print(f"Loading fixed mask from {mask_path}")
-        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
+        mask_loaded_image = cv2.imread(mask_path)
+        mask_loaded_image = cv2.cvtColor(mask_loaded_image, cv2.COLOR_BGR2RGB)
+        mask_loaded_image = cv2.resize(mask_loaded_image, (resolution, resolution), interpolation=cv2.INTER_AREA) / 255.0
+        mask_loaded_image = rearrange(torch.from_numpy(mask_loaded_image), "h w c -> c h w")
+        return mask_loaded_image
 
     def detect_facial_landmarks(self, image: np.ndarray):
         height, width, _ = image.shape