Change absolute path to relative path

ariel415el · Apr 26, 2022 · 690a58a · 690a58a
1 parent ac4fbc5
commit 690a58a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/scripts/reshuffle.py b/scripts/reshuffle.py
@@ -1,5 +1,6 @@
 import sys
 import os
+from pathlib import Path
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
 import distribution_metrics
@@ -23,6 +24,7 @@ def reshuffle(dataset_dir, out_dir, n_reps=1):
 
 if __name__ == '__main__':
     out_dir = "outputs/reshuffle"
-    reshuffle('/home/ariel/university/GPDM/images/SIGD16', out_dir)
-    reshuffle('/home/ariel/university/GPDM/images/Places50', out_dir)
-    reshuffle('/home/ariel/university/GPDM/images/reshuffle', out_dir)
+    base_dir = Path(__file__).parent.parent
+    reshuffle(base_dir / 'images/SIGD16', out_dir)
+    reshuffle(base_dir / 'images/Places50', out_dir)
+    reshuffle(base_dir / 'images/reshuffle', out_dir)
diff --git a/scripts/reshuffle_HD_image.py b/scripts/reshuffle_HD_image.py
@@ -1,5 +1,6 @@
 import sys
 import os
+from pathlib import Path
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
 import distribution_metrics
@@ -27,7 +28,8 @@ def reshuffle_hq(image_paths, out_dir, n_reps=1):
 if __name__ == '__main__':
 
     image_paths = []
-    dataset_dir = '/home/ariel/university/GPDM/images/HQ_16'
+    base_dir = Path(__file__).parent.parent
+    dataset_dir = base_dir / 'images/HQ_16'
     image_paths += [os.path.join(dataset_dir, x) for x in os.listdir(dataset_dir)]
     image_paths = [x for x in image_paths if ('rio' in x or 'lambs' in x or 'safari' in x)]