Spaces:
Sleeping
Sleeping
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'):
|