Code changes
Browse files- README.md +1 -1
- requirements.txt +4 -0
README.md
CHANGED
|
@@ -13,7 +13,7 @@ pretty_name: 3D Brain Structure MRI PCA
|
|
| 13 |
|
| 14 |
## 🧠 Model Summary
|
| 15 |
# brain2vec
|
| 16 |
-
|
| 17 |
|
| 18 |
|
| 19 |
# Training data
|
|
|
|
| 13 |
|
| 14 |
## 🧠 Model Summary
|
| 15 |
# brain2vec
|
| 16 |
+
A linear PCA model for brain structure T1 MRIs. The models takes in a 3d MRI NIfTI file and compresses to 1200 latent dimensions before reconstructing the image.
|
| 17 |
|
| 18 |
|
| 19 |
# Training data
|
requirements.txt
CHANGED
|
@@ -7,10 +7,14 @@ torch>=1.12
|
|
| 7 |
monai-weekly
|
| 8 |
monai-generative
|
| 9 |
|
|
|
|
|
|
|
|
|
|
| 10 |
# Common Python libraries
|
| 11 |
pandas
|
| 12 |
numpy
|
| 13 |
nibabel
|
|
|
|
| 14 |
matplotlib
|
| 15 |
datasets
|
| 16 |
scikit-learn
|
|
|
|
| 7 |
monai-weekly
|
| 8 |
monai-generative
|
| 9 |
|
| 10 |
+
# For perceptual losses in MONAI's generative module.
|
| 11 |
+
lpips
|
| 12 |
+
|
| 13 |
# Common Python libraries
|
| 14 |
pandas
|
| 15 |
numpy
|
| 16 |
nibabel
|
| 17 |
+
tqdm
|
| 18 |
matplotlib
|
| 19 |
datasets
|
| 20 |
scikit-learn
|