Update videoretalking/inference_function.py

videoretalking/inference_function.py

@@ -9,18 +9,18 @@ sys.path.insert(0, 'third_part')
 sys.path.insert(0, 'third_part/GPEN')
 
 # 3dmm extraction
-from third_part.face3d.util.preprocess import align_img
-from third_part.face3d.util.load_mats import load_lm3d
-from third_part.face3d.extract_kp_videos import KeypointExtractor
+from .third_part.face3d.util.preprocess import align_img
+from .third_part.face3d.util.load_mats import load_lm3d
+from .third_part.face3d.extract_kp_videos import KeypointExtractor
 # face enhancement
-from third_part.GPEN.gpen_face_enhancer import FaceEnhancement
+from .third_part.GPEN.gpen_face_enhancer import FaceEnhancement
 # # expression control
 # from third_part.ganimation_replicate.model.ganimation import GANimationModel
 
-from utils import audio
-from utils.ffhq_preprocess import Croper
-from utils.alignment_stit import crop_faces, calc_alignment_coefficients, paste_image
-from utils.inference_utils import Laplacian_Pyramid_Blending_with_mask, face_detect, load_model, options, split_coeff, \
+from .utils import audio
+from .utils.ffhq_preprocess import Croper
+from .utils.alignment_stit import crop_faces, calc_alignment_coefficients, paste_image
+from .utils.inference_utils import Laplacian_Pyramid_Blending_with_mask, face_detect, load_model, options, split_coeff, \
                                   trans_image, transform_semantic, find_crop_norm_ratio, load_face3d_net, exp_aus_dict
 import warnings
 warnings.filterwarnings("ignore")