This commit is contained in:
PowerHouseMan
2024-08-02 21:40:22 +09:00
parent 1e28e2d948
commit 3897947eeb
4 changed files with 5 additions and 5 deletions
+1 -1
View File
@@ -7,7 +7,7 @@ import yaml
from LivePortrait.src.utils.timer import Timer
from LivePortrait.src.utils.helper import load_model, concat_feat
from LivePortrait.src.utils.camera import headpose_pred_to_degree, get_rotation_matrix
from config.inference_config import InferenceConfig
from LivePortrait.src.config.inference_config import InferenceConfig
from LivePortrait.src.utils.rprint import rlog as log
+4 -4
View File
@@ -19,10 +19,10 @@ from scipy.spatial.transform import Rotation as R
current_file_path = os.path.abspath(__file__)
current_directory = os.path.dirname(current_file_path)
sys.path.append(current_directory)
from .LivePortrait.src.live_portrait_wrapper import LivePortraitWrapper
from .LivePortrait.src.utils.rprint import rlog as log
from .LivePortrait.src.utils.camera import get_rotation_matrix
from .config.inference_config import InferenceConfig
from LivePortrait.src.live_portrait_wrapper import LivePortraitWrapper
from LivePortrait.src.utils.rprint import rlog as log
from LivePortrait.src.utils.camera import get_rotation_matrix
from LivePortrait.src.config.inference_config import InferenceConfig
def tensor2pil(image):
return Image.fromarray(np.clip(255. * image.cpu().numpy().squeeze(), 0, 255).astype(np.uint8))