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
⚠️ Each model should be used with paired checkpoint and text encoder checkpoint.