From 3897947eebb71a37aba70b3bb7775cf92e0630f9 Mon Sep 17 00:00:00 2001 From: PowerHouseMan Date: Fri, 2 Aug 2024 21:40:22 +0900 Subject: [PATCH] main --- {config => LivePortrait/src/config}/inference_config.py | 0 {config => LivePortrait/src/config}/models.yaml | 0 LivePortrait/src/live_portrait_wrapper.py | 2 +- nodes.py | 8 ++++---- 4 files changed, 5 insertions(+), 5 deletions(-) rename {config => LivePortrait/src/config}/inference_config.py (100%) rename {config => LivePortrait/src/config}/models.yaml (100%) diff --git a/config/inference_config.py b/LivePortrait/src/config/inference_config.py similarity index 100% rename from config/inference_config.py rename to LivePortrait/src/config/inference_config.py diff --git a/config/models.yaml b/LivePortrait/src/config/models.yaml similarity index 100% rename from config/models.yaml rename to LivePortrait/src/config/models.yaml diff --git a/LivePortrait/src/live_portrait_wrapper.py b/LivePortrait/src/live_portrait_wrapper.py index 935b03e..ed19753 100644 --- a/LivePortrait/src/live_portrait_wrapper.py +++ b/LivePortrait/src/live_portrait_wrapper.py @@ -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 diff --git a/nodes.py b/nodes.py index 6482d4a..eac17b9 100644 --- a/nodes.py +++ b/nodes.py @@ -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))