jhj0517 commited on
Commit
6058d6e
1 Parent(s): c020be9

Add video output path

Browse files
Files changed (1) hide show
  1. modules/utils/paths.py +4 -1
modules/utils/paths.py CHANGED
@@ -6,6 +6,7 @@ PROJECT_ROOT_DIR = os.path.join(os.path.abspath(os.path.dirname(__file__)), ".."
6
  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
  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")
@@ -53,7 +54,9 @@ def init_dirs():
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
 
 
6
  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
+ OUTPUTS_VIDEOS_DIR = os.path.join(OUTPUTS_DIR, "videos")
10
  TEMP_DIR = os.path.join(OUTPUTS_DIR, "temp")
11
  TEMP_VIDEO_FRAMES_DIR = os.path.join(TEMP_DIR, "video_frames")
12
  TEMP_VIDEO_OUT_FRAMES_DIR = os.path.join(TEMP_VIDEO_FRAMES_DIR, "output_video_frames")
 
54
  OUTPUTS_DIR,
55
  EXP_OUTPUT_DIR,
56
  TEMP_DIR,
57
+ TEMP_VIDEO_FRAMES_DIR,
58
+ TEMP_VIDEO_OUT_FRAMES_DIR,
59
+ OUTPUTS_VIDEOS_DIR
60
  ]:
61
  os.makedirs(dir_path, exist_ok=True)
62