samarth-ht commited on
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)