Skip to content

Commit 8dd1d8c

Browse files
committed
[+] add checkpoints
1 parent 4e45d3d commit 8dd1d8c

File tree

1 file changed

+35
-0
lines changed

1 file changed

+35
-0
lines changed

resources/resources.md

Lines changed: 35 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,3 +3,38 @@
33
## TCIA Dataset
44

55
## TotalSegmentator Dataset
6+
7+
## Checkpoints
8+
9+
We provide two model checkpoints that you can use directly with nnU-Net:
10+
11+
- **mixed_model_continual_learning.zip**
12+
13+
- **pure_pediatric_model.zip**
14+
15+
### Installing Pretrained Models
16+
17+
nnU-Net offers a convenient utility to install pretrained models from a zip archive. Follow these steps:
18+
19+
1. **Downloading Pretrained Weights:**
20+
Go to the [GitHub Releases](https://github.com/ICANS-Strasbourg/PSAT/releases) page.
21+
22+
Download:
23+
- `mixed_model_continual_learning.zip`
24+
- `pure_pediatric_model.zip`
25+
26+
Place them in `resources/checkpoints/`.
27+
28+
2. **Install the Checkpoint Using nnU-Net**
29+
Use the `nnUNet_install_pretrained_model_from_zip` command in your terminal.
30+
31+
```bash
32+
# To install the mixed model checkpoint:
33+
nnUNetv2_install_pretrained_model_from_zip resources/checkpoints/mixed_model_continual_learning.zip
34+
35+
# To install the pure pediatric model checkpoint:
36+
nnUNetv2_install_pretrained_model_from_zip resources/checkpoints/pure_pediatric_model.zip
37+
38+
3. **Running Inference**
39+
40+
Once you have installed a checkpoint, you can run inference on your input images using the `nnUNet_predict` command.

0 commit comments

Comments
 (0)