update project struct

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

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",

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

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)