Spaces:
Running
Running
gavinyuan
commited on
Commit
·
fbf97aa
1
Parent(s):
5c64773
update: GPEN weights
Browse files- inference/tricks.py +1 -1
inference/tricks.py
CHANGED
|
@@ -52,7 +52,7 @@ class Trick(object):
|
|
| 52 |
blur_size = tuple(2 * j + 1 for j in kernel_size)
|
| 53 |
facial_mask = cv2.GaussianBlur(facial_mask, blur_size, 0)
|
| 54 |
|
| 55 |
-
return facial_mask.astype(np.
|
| 56 |
|
| 57 |
@staticmethod
|
| 58 |
def smooth_mask(mask_tensor: torch.Tensor):
|
|
|
|
| 52 |
blur_size = tuple(2 * j + 1 for j in kernel_size)
|
| 53 |
facial_mask = cv2.GaussianBlur(facial_mask, blur_size, 0)
|
| 54 |
|
| 55 |
+
return facial_mask.astype(np.float32) / 255
|
| 56 |
|
| 57 |
@staticmethod
|
| 58 |
def smooth_mask(mask_tensor: torch.Tensor):
|