File size: 656 Bytes
e30b1cc fa74b5c e30b1cc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
This repo contains the checkpoints for SAT.
We offer SAT-Pro, SAT-Nano (both trained on 72 datasets) and another 5 different variants of SAT-Nano (all trained on 49 datasets):
- SAT-Pro: ./Pro
- SAT-Nano: ./Nano
- UNET-Ours: ./Others/UNET-Ours
- UNET-CPT: ./Others/UNET-CPT
- UNET-BB: ./Others/UNET-BaseBERT
- UMamba-CPT: ./Others/UMamba-CPT
- SwinUNETR-CPT: ./Others/SwinUNETR-CPT
Check our [paper](https://github.com/zhaoziheng/SAT/tree/main) for more details, and [github repo](https://github.com/zhaoziheng/SAT/tree/main?tab=readme-ov-file) for usage instruction.
⚠️ Each model should be used with paired checkpoint and text encoder checkpoint. |