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.