nnunetv2
medical-imaging
radiology
3d
self-supervised
foundation-model
nnunet
nnssl
consti95 commited on
Commit
c691f21
·
verified ·
1 Parent(s): 29b9041

Fix model pair table names; add classification fine-tuning link

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -26,7 +26,7 @@ with the [`nnssl`](https://github.com/MIC-DKFZ/nnssl) self-supervised learning f
26
 
27
  | Model | Architecture | Details | Params | Repository |
28
  |---|---|---|---|---|
29
- | nnFoundationViT | [ResEnc](https://arxiv.org/abs/2404.09556) | 6 stages, features 32–64–128–256–320–320 | 102M | [MIC-DKFZ/nnFoundationCNN](https://huggingface.co/MIC-DKFZ/nnFoundationCNN) |
30
  | **nnFoundationViT** | [Primus](https://arxiv.org/abs/2503.01835) | 40 layers, embedding dim 1056, 16 heads, 8³ patch tokens | 674M | this repository |
31
 
32
  ## Using nnFoundation
@@ -35,7 +35,7 @@ To fine-tune nnFoundation on your own downstream tasks, use one of our dedicated
35
 
36
  - **Segmentation:** [nnU-Net -- Fine-tuning from nnssl checkpoints](https://github.com/MIC-DKFZ/nnUNet/blob/master/documentation/finetuning_from_nnssl_checkpoints.md)
37
  - **Detection:** [nnDetection -- Fine-tuning](https://github.com/MIC-DKFZ/nnDetection/blob/finetuning/docs/finetuning.md)
38
- - **Classification:** TBA
39
  - **Report generation:** TBA
40
 
41
  ### Planning and preprocessing straight from this repository
 
26
 
27
  | Model | Architecture | Details | Params | Repository |
28
  |---|---|---|---|---|
29
+ | nnFoundationCNN | [ResEnc](https://arxiv.org/abs/2404.09556) | 6 stages, features 32–64–128–256–320–320 | 102M | [MIC-DKFZ/nnFoundationCNN](https://huggingface.co/MIC-DKFZ/nnFoundationCNN) |
30
  | **nnFoundationViT** | [Primus](https://arxiv.org/abs/2503.01835) | 40 layers, embedding dim 1056, 16 heads, 8³ patch tokens | 674M | this repository |
31
 
32
  ## Using nnFoundation
 
35
 
36
  - **Segmentation:** [nnU-Net -- Fine-tuning from nnssl checkpoints](https://github.com/MIC-DKFZ/nnUNet/blob/master/documentation/finetuning_from_nnssl_checkpoints.md)
37
  - **Detection:** [nnDetection -- Fine-tuning](https://github.com/MIC-DKFZ/nnDetection/blob/finetuning/docs/finetuning.md)
38
+ - **Classification:** [image_classification -- Fine-tuning nnFoundation](https://github.com/MIC-DKFZ/image_classification/blob/master/docs/nnfoundation.md)
39
  - **Report generation:** TBA
40
 
41
  ### Planning and preprocessing straight from this repository