update colab notebook

huzeyann · Feb 7, 2024 · 8f16e48 · 8f16e48
1 parent 6896ae5
commit 8f16e48
Show file tree

Hide file tree

Showing 3 changed files with 233 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -2,7 +2,7 @@
 ## Brain Decodes Deep Nets <br><sub> PyTorch Implementation</sub>
 
 
-### [📄Paper](https://arxiv.org/abs/2312.01280) | [🌐Website](https://huzeyann.github.io/brain-decodes-deep-nets)
+### [📄Paper](https://arxiv.org/abs/2312.01280) | [🌐Website](https://huzeyann.github.io/brain-decodes-deep-nets) | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://drive.google.com/file/d/158tm5AvCc346SDM1WqxanRjqcJFh5k5X/view?usp=sharing]
 
 We visualize pre-trained vision models by mapping them onto the brain, thus exposing their hidden inside. Visualization is a by-product of brain encoding model: predict brain fMRI measurements in response to images. 
 

diff --git a/brainnet/plot_utils.py b/brainnet/plot_utils.py
@@ -273,7 +273,7 @@ def make_space_plot(mu, path):
     )
 
     dir_path = os.path.dirname(path)
-    os.makedirs(dir_path, exist_ok=True)
+    # os.makedirs(dir_path, exist_ok=True)
 
     plt.close()
 
@@ -324,7 +324,7 @@ def rgb_from_values(values, vmax=1):
     )
     COLORFUL_PLOT = False
     dir_path = os.path.dirname(path)
-    os.makedirs(dir_path, exist_ok=True)
+    # os.makedirs(dir_path, exist_ok=True)
 
     plt.close()
 
@@ -365,7 +365,7 @@ def make_1d_plot(
     )
     dir_path = os.path.dirname(path)
     # plt.show()
-    os.makedirs(dir_path, exist_ok=True)
+    # os.makedirs(dir_path, exist_ok=True)
     plt.close()
 
     center_crop(path, overlay=overlay)

diff --git a/inference.ipynb b/inference.ipynb