gavinyuan commited on
Commit
fbf97aa
1 Parent(s): 5c64773

update: GPEN weights

Browse files
Files changed (1) hide show
  1. 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.float) / 255
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):