Update model link to Zenodo

fmcib/run.py

@@ -11,7 +11,7 @@
 def get_features(
     csv_path,
     weights_path=None,
-    download_url="https://www.dropbox.com/s/bd7azdsvx1jhalp/fmcib.torch?dl=1",
+    download_url="https://zenodo.org/records/10528450/files/model_weights.torch?download=1",
     spatial_size=(50, 50, 50),
     precropped=False,
 ):
@@ -21,7 +21,7 @@ def get_features(
     Args:
         csv_path (str): Path to the CSV file containing image paths.
         weights_path (str, optional): Path to the pre-trained weights file. Defaults to None.
-        download_url (str, optional): URL to download the pre-trained weights file. Defaults to "https://www.dropbox.com/s/bd7azdsvx1jhalp/fmcib.torch?dl=1".
+        download_url (str, optional): URL to download the pre-trained weights file. Defaults to "https://zenodo.org/records/10528450/files/model_weights.torch?download=1".
         spatial_size (tuple, optional): Spatial size of the input images. Defaults to (50, 50, 50).
         precropped (bool, optional): Whether the images are already pre-cropped. Defaults to False.
 

models/download_foundation_pretrained_model.sh

@@ -0,0 +1,2 @@
+mkdir ssl
+wget https://zenodo.org/records/10528450/files/model_weights.torch?download=1 -O ssl/fmcib.torch

models/download_task1_baselines.sh

@@ -0,0 +1,11 @@
+if [ ! -d "supervised" ]; then
+  mkdir supervised
+fi
+
+cd "supervised"
+if [ ! -d "task1" ]; then
+  mkdir task1
+fi
+
+wget https://www.dropbox.com/s/a7sjdigfa29355y/task1_foundation_finetuned.torch?dl=1 -O supervised/task1_foundation_finetuned.torch
+wget https://www.dropbox.com/s/bo0scqq1g80li22/task1_supervised.torch?dl=1 -O supervised/task1_supervised.torch

models/download_task2_baselines.sh

@@ -0,0 +1,12 @@
+if [ ! -d "supervised" ]; then
+  mkdir supervised
+fi
+
+cd "supervised"
+if [ ! -d "task2" ]; then
+  mkdir task2
+fi
+
+wget https://www.dropbox.com/s/zypjv5k6wohxwn3/task2_foundation_finetuned.torch?dl=1 -O supervised/task2_foundation_finetuned.torch
+wget https://www.dropbox.com/s/7pu0irhlubcp6ji/task2_supervised_finetuned.torch?dl=1 -O supervised/task2_supervised_finetuned.torch
+wget https://www.dropbox.com/s/44f7stxvxu9n2d1/task2_supervised.torch?dl=1 -O supervised/task2_supervised.torch

models/download_task3_baselines.sh

@@ -0,0 +1,12 @@
+if [ ! -d "supervised" ]; then
+  mkdir supervised
+fi
+
+cd "supervised"
+if [ ! -d "task3" ]; then
+  mkdir task3
+fi
+
+wget https://www.dropbox.com/s/u6uulvijpkd9mf6/task3_foundation_finetuned.torch?dl=1 -O supervised/task3_foundation_finetuned.torch
+wget https://www.dropbox.com/s/94512wr5lmhwqgc/task3_supervised_finetuned.torch?dl=1 -O supervised/task3_supervised_finetuned.torch
+wget https://www.dropbox.com/s/57ga9isdduz4ea6/task3_supervised.torch?dl=1 -O supervised/task3_supervised.torch