jhj0517
commited on
Commit
·
f2aa528
1
Parent(s):
f2a34e7
Add exp output dir
Browse files- modules/utils/paths.py +4 -1
modules/utils/paths.py
CHANGED
@@ -5,6 +5,7 @@ import os
|
|
5 |
PROJECT_ROOT_DIR = os.path.join(os.path.abspath(os.path.dirname(__file__)), "..", "..")
|
6 |
MODELS_DIR = os.path.join(PROJECT_ROOT_DIR, "models")
|
7 |
OUTPUTS_DIR = os.path.join(PROJECT_ROOT_DIR, "outputs")
|
|
|
8 |
MODEL_CONFIG = os.path.join(PROJECT_ROOT_DIR, "modules", "config", "models.yaml")
|
9 |
MODEL_PATHS = {
|
10 |
"appearance_feature_extractor": os.path.join(MODELS_DIR, "appearance_feature_extractor.safetensors"),
|
@@ -14,13 +15,15 @@ MODEL_PATHS = {
|
|
14 |
"stitching_retargeting_module": os.path.join(MODELS_DIR, "stitching_retargeting_module.safetensors"),
|
15 |
"face_yolov8n": os.path.join(MODELS_DIR, "face_yolov8n.pt")
|
16 |
}
|
|
|
17 |
|
18 |
|
19 |
@functools.lru_cache
|
20 |
def init_dirs():
|
21 |
for dir_path in [
|
22 |
MODELS_DIR,
|
23 |
-
OUTPUTS_DIR
|
|
|
24 |
]:
|
25 |
os.makedirs(dir_path, exist_ok=True)
|
26 |
|
|
|
5 |
PROJECT_ROOT_DIR = os.path.join(os.path.abspath(os.path.dirname(__file__)), "..", "..")
|
6 |
MODELS_DIR = os.path.join(PROJECT_ROOT_DIR, "models")
|
7 |
OUTPUTS_DIR = os.path.join(PROJECT_ROOT_DIR, "outputs")
|
8 |
+
EXP_OUTPUT_DIR = os.path.join(OUTPUTS_DIR, "exp_data")
|
9 |
MODEL_CONFIG = os.path.join(PROJECT_ROOT_DIR, "modules", "config", "models.yaml")
|
10 |
MODEL_PATHS = {
|
11 |
"appearance_feature_extractor": os.path.join(MODELS_DIR, "appearance_feature_extractor.safetensors"),
|
|
|
15 |
"stitching_retargeting_module": os.path.join(MODELS_DIR, "stitching_retargeting_module.safetensors"),
|
16 |
"face_yolov8n": os.path.join(MODELS_DIR, "face_yolov8n.pt")
|
17 |
}
|
18 |
+
MASK_TEMPLATES = os.path.join(PROJECT_ROOT_DIR, "modules", "utils", "resources", "mask_template.png")
|
19 |
|
20 |
|
21 |
@functools.lru_cache
|
22 |
def init_dirs():
|
23 |
for dir_path in [
|
24 |
MODELS_DIR,
|
25 |
+
OUTPUTS_DIR,
|
26 |
+
EXP_OUTPUT_DIR
|
27 |
]:
|
28 |
os.makedirs(dir_path, exist_ok=True)
|
29 |
|