| 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. |