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))