add download from ixi and caching dataset.

INFN-MRI · Dec 5, 2024 · be807df · be807df
1 parent 42a7eae
commit be807df
Show file tree

Hide file tree

Showing 3 changed files with 39 additions and 11 deletions.
diff --git a/.github/workflows/test-ci.yml b/.github/workflows/test-ci.yml
@@ -147,6 +147,13 @@ jobs:
         uses: actions/setup-python@v4
         with:
           python-version: "3.10"
+
+      - name: Restore cached ixi directory
+        uses: actions/cache/restore@v4
+        id: cache-restore
+        with:
+          path: ~/.ixi
+          key: ${{ runner.os }}-ixi
 
       - name: Install dependencies
         shell: bash -l {0}
@@ -162,6 +169,13 @@ jobs:
       - name: Display data
         run: ls -R
         working-directory: docs_build/_static
+
+      - name: Cache ixi directory
+        uses: actions/cache/save@v4
+        if: ${{ steps.cache-restore.outputs.cache-hit != 'true' }}
+        with:
+          path: ~/.ixi
+          key: ${{ runner.os }}-ixi
 
       - name: Upload artifact
         id: artifact-upload-step

diff --git a/examples/02-synth-data.py b/examples/02-synth-data.py
@@ -30,12 +30,19 @@
 
 import torchio as tio
 
-ixi_dataset = tio.datasets.IXI(
-    "/home/mcencini//ixi/",
-    modalities=("PD", "T1", "T2"),
-    download=False,
-)
-
+try:
+    ixi_dataset = tio.datasets.IXI(
+        "/home/mcencini/ixi/",
+        modalities=("PD", "T1", "T2"),
+        download=False,
+    )
+except:
+    ixi_dataset = tio.datasets.IXI(
+        "$HOME/.ixi/",
+        modalities=("PD", "T1", "T2"),
+        download=True,
+    )
+
 # get subject 0
 sample_subject = ixi_dataset[0]
 

diff --git a/examples/03-fitting.py b/examples/03-fitting.py
@@ -30,11 +30,18 @@
 import torchsim
 
 
-ixi_dataset = tio.datasets.IXI(
-    "/home/mcencini//ixi/",
-    modalities=("PD", "T1", "T2"),
-    download=False,
-)
+try:
+    ixi_dataset = tio.datasets.IXI(
+        "/home/mcencini/ixi/",
+        modalities=("PD", "T1", "T2"),
+        download=False,
+    )
+except:
+    ixi_dataset = tio.datasets.IXI(
+        "$HOME/.ixi/",
+        modalities=("PD", "T1", "T2"),
+        download=True,
+    )
 
 # get subject 0
 sample_subject = ixi_dataset[0]