diff --git a/__init__.py b/__init__.py index 495d627..143d69a 100644 --- a/__init__.py +++ b/__init__.py @@ -3,14 +3,14 @@ import os main_path = os.path.dirname(__file__) sys.path.append(main_path) -from protrait.config import * +from portrait.config import * import subprocess import threading import requests from tqdm import tqdm -from protrait.nodes import * +from portrait.nodes import * # import pydevd_pycharm # pydevd_pycharm.settrace('49.7.62.197', port=10090, stdoutToServer=True, stderrToServer=True) diff --git a/protrait/__init__.py b/portrait/__init__.py similarity index 100% rename from protrait/__init__.py rename to portrait/__init__.py diff --git a/protrait/config.py b/portrait/config.py similarity index 94% rename from protrait/config.py rename to portrait/config.py index 4d40b85..dd8fc4b 100644 --- a/protrait/config.py +++ b/portrait/config.py @@ -1,9 +1,9 @@ import os from folder_paths import folder_names_and_paths +from .. import main_path -root_path = os.path.dirname(__file__) -utils_path = os.path.join(os.path.dirname(__file__), "utils") -models_path = os.path.join(os.path.dirname(__file__), "models") +utils_path = os.path.join(main_path, "utils") +models_path = os.path.join(main_path, "models") # save_dirs urls = [ "https://pai-aigc-photog.oss-cn-hangzhou.aliyuncs.com/webui/ChilloutMix-ni-fp16.safetensors", diff --git a/protrait/nodes.py b/portrait/nodes.py similarity index 98% rename from protrait/nodes.py rename to portrait/nodes.py index af61a2c..e01c96f 100644 --- a/protrait/nodes.py +++ b/portrait/nodes.py @@ -6,8 +6,8 @@ from PIL import Image from modelscope.outputs import OutputKeys from modelscope.pipelines import pipeline from modelscope.utils.constant import Tasks -from utils.face_process_utils import call_face_crop, color_transfer, Face_Skin -from utils.img_utils import img_to_tensor, tensor_to_img, tensor_to_np, np_to_tensor, np_to_mask, img_to_mask +from .utils.face_process_utils import call_face_crop, color_transfer, Face_Skin +from .utils.img_utils import img_to_tensor, tensor_to_img, tensor_to_np, np_to_tensor, np_to_mask, img_to_mask import insightface from insightface.app import FaceAnalysis from insightface.data import get_image as ins_get_image diff --git a/protrait/utils/__init__.py b/portrait/utils/__init__.py similarity index 100% rename from protrait/utils/__init__.py rename to portrait/utils/__init__.py diff --git a/protrait/utils/face_process_utils.py b/portrait/utils/face_process_utils.py similarity index 99% rename from protrait/utils/face_process_utils.py rename to portrait/utils/face_process_utils.py index 01f6a5a..5197b0e 100644 --- a/protrait/utils/face_process_utils.py +++ b/portrait/utils/face_process_utils.py @@ -6,7 +6,7 @@ import torch.nn.functional as F import torchvision.transforms as transforms from PIL import Image from skimage import transform -from img_utils import np_to_mask +from .img_utils import np_to_mask # import pydevd_pycharm # pydevd_pycharm.settrace('49.7.62.197', port=10090, stdoutToServer=True, stderrToServer=True) diff --git a/protrait/utils/img_utils.py b/portrait/utils/img_utils.py similarity index 100% rename from protrait/utils/img_utils.py rename to portrait/utils/img_utils.py