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