Update README.md
Browse files
README.md
CHANGED
|
@@ -21,7 +21,11 @@ For more information on our method, we refer to our [**project page**](https://p
|
|
| 21 |
If you want to use the pre-trained models provided in this repository, download the model weights and follow the instructions in the official [GitHub repository](https://github.com/pfriedri/wdm-3d).
|
| 22 |
|
| 23 |
## Pre-trained models
|
| 24 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 25 |
|
| 26 |
## Hardware requirements
|
| 27 |
To sample images from the provided models, you require a GPU with at least:
|
|
|
|
| 21 |
If you want to use the pre-trained models provided in this repository, download the model weights and follow the instructions in the official [GitHub repository](https://github.com/pfriedri/wdm-3d).
|
| 22 |
|
| 23 |
## Pre-trained models
|
| 24 |
+
### BraTS 2023 (T1-weighted brain MR image generation)
|
| 25 |
+
- [Download model](brats_unet_128_1200k.pt) Resolution: 128 x 128 x 128, Backbone: U-Net, Trained: 1.2M iterations
|
| 26 |
+
|
| 27 |
+
### LIDC-IDRI (Lung CT image generation)
|
| 28 |
+
- [Download model](lidc-idri_128_unet_1200k.pt) Resolution: 128 x 128 x 128, Backbone: U-Net, Trained: 1.2M iterations
|
| 29 |
|
| 30 |
## Hardware requirements
|
| 31 |
To sample images from the provided models, you require a GPU with at least:
|