|
import functools |
|
import os |
|
|
|
|
|
PROJECT_ROOT_DIR = os.path.join(os.path.abspath(os.path.dirname(__file__)), "..", "..") |
|
MODELS_DIR = os.path.join(PROJECT_ROOT_DIR, "models") |
|
OUTPUTS_DIR = os.path.join(PROJECT_ROOT_DIR, "outputs") |
|
EXP_OUTPUT_DIR = os.path.join(OUTPUTS_DIR, "exp_data") |
|
MODEL_CONFIG = os.path.join(PROJECT_ROOT_DIR, "modules", "config", "models.yaml") |
|
MODEL_PATHS = { |
|
"appearance_feature_extractor": os.path.join(MODELS_DIR, "appearance_feature_extractor.safetensors"), |
|
"motion_extractor": os.path.join(MODELS_DIR, "motion_extractor.safetensors"), |
|
"warping_module": os.path.join(MODELS_DIR, "warping_module.safetensors"), |
|
"spade_generator": os.path.join(MODELS_DIR, "spade_generator.safetensors"), |
|
"stitching_retargeting_module": os.path.join(MODELS_DIR, "stitching_retargeting_module.safetensors"), |
|
"face_yolov8n": os.path.join(MODELS_DIR, "face_yolov8n.pt") |
|
} |
|
MASK_TEMPLATES = os.path.join(PROJECT_ROOT_DIR, "modules", "utils", "resources", "mask_template.png") |
|
|
|
|
|
@functools.lru_cache |
|
def init_dirs(): |
|
for dir_path in [ |
|
MODELS_DIR, |
|
OUTPUTS_DIR, |
|
EXP_OUTPUT_DIR |
|
]: |
|
os.makedirs(dir_path, exist_ok=True) |
|
|
|
|
|
init_dirs() |
|
|
