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