Hugging Face's logo

pfriedri
/
wdm-3d

English
3d
medical
image-synthesis
image-generation
wavelet-transform
Model card Files Community
pfriedri commited on
Commit
9d01ae5
•
1 Parent(s): e2fa221

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -1
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
- We will soon provide models for different datasets and different image resolutions (starting with BraTs and LIDC-IDRI at resolutions of 128³ and 256³).
 
 
 
 
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: