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