jhj0517 commited on
Commit
e492152
·
1 Parent(s): 0111784

Add temp video frames dir path

Browse files
Files changed (1) hide show
  1. modules/utils/paths.py +4 -1
modules/utils/paths.py CHANGED
@@ -7,6 +7,8 @@ MODELS_DIR = os.path.join(PROJECT_ROOT_DIR, "models")
7
  MODELS_ANIMAL_DIR = os.path.join(MODELS_DIR, "animal")
8
  OUTPUTS_DIR = os.path.join(PROJECT_ROOT_DIR, "outputs")
9
  TEMP_DIR = os.path.join(OUTPUTS_DIR, "temp")
 
 
10
  EXP_OUTPUT_DIR = os.path.join(OUTPUTS_DIR, "exp_data")
11
  MODEL_CONFIG = os.path.join(PROJECT_ROOT_DIR, "modules", "config", "models.yaml")
12
  MODEL_PATHS = {
@@ -50,7 +52,8 @@ def init_dirs():
50
  MODELS_ANIMAL_DIR,
51
  OUTPUTS_DIR,
52
  EXP_OUTPUT_DIR,
53
- TEMP_DIR
 
54
  ]:
55
  os.makedirs(dir_path, exist_ok=True)
56
 
 
7
  MODELS_ANIMAL_DIR = os.path.join(MODELS_DIR, "animal")
8
  OUTPUTS_DIR = os.path.join(PROJECT_ROOT_DIR, "outputs")
9
  TEMP_DIR = os.path.join(OUTPUTS_DIR, "temp")
10
+ TEMP_VIDEO_FRAMES_DIR = os.path.join(TEMP_DIR, "video_frames")
11
+ TEMP_VIDEO_OUT_FRAMES_DIR = os.path.join(TEMP_VIDEO_FRAMES_DIR, "output_video_frames")
12
  EXP_OUTPUT_DIR = os.path.join(OUTPUTS_DIR, "exp_data")
13
  MODEL_CONFIG = os.path.join(PROJECT_ROOT_DIR, "modules", "config", "models.yaml")
14
  MODEL_PATHS = {
 
52
  MODELS_ANIMAL_DIR,
53
  OUTPUTS_DIR,
54
  EXP_OUTPUT_DIR,
55
+ TEMP_DIR,
56
+ TEMP_VIDEO_FRAMES_DIR
57
  ]:
58
  os.makedirs(dir_path, exist_ok=True)
59