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