Spaces:
Sleeping
Sleeping
Commit
·
a9d463f
1
Parent(s):
eca4597
class error resolve
Browse files
soundimage/utils/image_processor.py
CHANGED
@@ -63,7 +63,7 @@ class ImageProcessor:
|
|
63 |
self.face_mesh = None
|
64 |
self.fa = None
|
65 |
|
66 |
-
def load_fixed_mask(resolution: int, mask_path: str) -> torch.Tensor:
|
67 |
print(f"Loading fixed mask from {mask_path}")
|
68 |
mask_loaded_image = cv2.imread(mask_path)
|
69 |
mask_loaded_image = cv2.cvtColor(mask_loaded_image, cv2.COLOR_BGR2RGB)
|
|
|
63 |
self.face_mesh = None
|
64 |
self.fa = None
|
65 |
|
66 |
+
def load_fixed_mask(self, resolution: int, mask_path: str) -> torch.Tensor:
|
67 |
print(f"Loading fixed mask from {mask_path}")
|
68 |
mask_loaded_image = cv2.imread(mask_path)
|
69 |
mask_loaded_image = cv2.cvtColor(mask_loaded_image, cv2.COLOR_BGR2RGB)
|