From 683c077f3deda8ff9c12084aa7154a62567e3b19 Mon Sep 17 00:00:00 2001 From: JianHe0628 <jl02958@surrey.ac.uk> Date: Sat, 14 Jun 2025 15:54:20 +0100 Subject: [PATCH] adding dynamic path switching --- hamer/configs/__init__.py | 5 ++++- vitpose_model.py | 8 +++++++- 2 files changed, 11 insertions(+), 2 deletions(-) diff --git a/hamer/configs/__init__.py b/hamer/configs/__init__.py index ebee64c..e42bfe7 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 5135ff1..656441a 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): -- GitLab