15 lines
514 B
Python
15 lines
514 B
Python
import os
|
|
|
|
# get the model paths
|
|
try:
|
|
from folder_paths import models_dir # pyright: ignore
|
|
except ImportError:
|
|
from pathlib import Path
|
|
|
|
models_dir = os.path.join(Path(__file__).parents[2], "models")
|
|
|
|
models_dir_path = os.path.join(models_dir, "onnx", "human-parts")
|
|
model_url = "https://huggingface.co/Metal3d/deeplabv3p-resnet50-human/resolve/main/deeplabv3p-resnet50-human.onnx"
|
|
model_name = os.path.basename(model_url)
|
|
model_path = os.path.join(models_dir_path, "deeplabv3p-resnet50-human.onnx")
|