Merge pull request #639 from yolain/modify_the_dir

Fix human segmentation bug

py/modules/human_parsing/parsing_api.py

@@ -4,7 +4,7 @@ import cv2
 import torchvision.transforms as transforms
 from torch.utils.data import DataLoader
 from .simple_extractor_dataset import SimpleFolderDataset
-from transforms import transform_logits
+from .transforms import transform_logits
 from tqdm import tqdm
 from PIL import Image
 

py/nodes/image.py

@@ -1348,7 +1348,7 @@ class humanSegmentation:
         if method in cache:
           _, parsing = cache[method][1]
         else:
-          from ..modules.human_parsing import HumanParsing
+          from ..modules.human_parsing.run_parsing import HumanParsing
           onnx_path = os.path.join(folder_paths.models_dir, 'onnx')
           model_path = get_local_filepath(HUMANPARSING_MODELS['parsing_lip']['model_url'], onnx_path)
           parsing = HumanParsing(model_path=model_path)