Ganrt commited on
Commit
da6d14f
1 Parent(s): f1de88d
third_part/GPEN/gpen_face_enhancer.py CHANGED
@@ -57,6 +57,7 @@ class FaceEnhancement(object):
57
  orig_faces, enhanced_faces = [], []
58
  height, width = img.shape[:2]
59
  full_mask = np.zeros((height, width), dtype=np.float32)
 
60
  full_img = np.zeros(ori_img.shape, dtype=np.uint8)
61
 
62
  for i, (faceb, facial5points) in enumerate(zip(facebs, landms)):
 
57
  orig_faces, enhanced_faces = [], []
58
  height, width = img.shape[:2]
59
  full_mask = np.zeros((height, width), dtype=np.float32)
60
+ mask_sharp = np.zeros((height, width), dtype=np.float32)
61
  full_img = np.zeros(ori_img.shape, dtype=np.uint8)
62
 
63
  for i, (faceb, facial5points) in enumerate(zip(facebs, landms)):