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):
|