diff --git a/hamer/configs/__init__.py b/hamer/configs/__init__.py index ebee64c7aed7819cf2443b3bc89fc1a3aa271bf8..e42bfe76c70715f557b1b5ddcf13e019bd5227e6 100644 --- a/hamer/configs/__init__.py +++ b/hamer/configs/__init__.py @@ -4,7 +4,10 @@ from yacs.config import CfgNode as CN # CACHE_DIR_HAMER = "./_DATA" # CACHE_DIR_HAMER = "/mnt/fast/nobackup/users/jl02958/hamer/_DATA" -CACHE_DIR_HAMER = "/vol/research/SignHamerFeatures/hamer/_DATA" +if os.path.exists("/vol/research/SignHamerFeatures/"): + CACHE_DIR_HAMER = "/vol/research/SignHamerFeatures/hamer/_DATA" +else: + CACHE_DIR_HAMER = "/mnt/fast/nobackup/users/jl02958/hamer/_DATA" def to_lower(x: Dict) -> Dict: """ diff --git a/vitpose_model.py b/vitpose_model.py index 5135ff19bef3d41520b79204dd4ce722ebc277bf..656441ae2ab86a9aee140834c83389f24e64a83a 100644 --- a/vitpose_model.py +++ b/vitpose_model.py @@ -13,7 +13,13 @@ os.environ["PYOPENGL_PLATFORM"] = "egl" # project root directory # ROOT_DIR = "/mnt/fast/nobackup/users/jl02958/hamer" -ROOT_DIR = "/vol/research/SignHamerFeatures/hamer" + +if os.path.exists("/vol/research/SignHamerFeatures/"): + ROOT_DIR = "/vol/research/SignHamerFeatures/hamer" +else: + ROOT_DIR = "/mnt/fast/nobackup/users/jl02958/hamer" + + VIT_DIR = os.path.join(ROOT_DIR, "third-party/ViTPose") class ViTPoseModel(object):