jhj0517 commited on
Commit
f2aa528
·
1 Parent(s): f2a34e7

Add exp output dir

Browse files
Files changed (1) hide show
  1. 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