Spanicin commited on
Commit
98eca54
1 Parent(s): 88de352

Update videoretalking/third_part/GPEN/gpen_face_enhancer.py

Browse files
videoretalking/third_part/GPEN/gpen_face_enhancer.py CHANGED
@@ -7,8 +7,8 @@ from videoretalking.third_part.GPEN.face_detect.retinaface_detection import Reti
7
  from videoretalking.third_part.GPEN.face_parse.face_parsing import FaceParse
8
  from videoretalking.third_part.GPEN.face_model.face_gan import FaceGAN
9
  # from sr_model.real_esrnet import RealESRNet
10
- from align_faces import warp_and_crop_face, get_reference_facial_points
11
- from utils.inference_utils import Laplacian_Pyramid_Blending_with_mask
12
 
13
  class FaceEnhancement(object):
14
  def __init__(self, base_dir='./', size=512, model=None, use_sr=True, sr_model=None, channel_multiplier=2, narrow=1, device='cuda'):
 
7
  from videoretalking.third_part.GPEN.face_parse.face_parsing import FaceParse
8
  from videoretalking.third_part.GPEN.face_model.face_gan import FaceGAN
9
  # from sr_model.real_esrnet import RealESRNet
10
+ from videoretalking.third_part.GPEN.align_faces import warp_and_crop_face, get_reference_facial_points
11
+ from videoretalking.utils.inference_utils import Laplacian_Pyramid_Blending_with_mask
12
 
13
  class FaceEnhancement(object):
14
  def __init__(self, base_dir='./', size=512, model=None, use_sr=True, sr_model=None, channel_multiplier=2, narrow=1, device='cuda'):