AFF-MAE released checkpoints
Inference weights for AFFMAE (ECCV 2026).
| file | stage | resolution | classes |
|---|---|---|---|
ckpt_epoch_399_affmae_fpw.pth |
pretrain (MAE) | 512 | - |
fpw_aff_base_ft_512_slits_pgbmi.pth |
segmentation | 512 | 3 |
fpw_aff_base_ft_768_slits_pgbmi.pth |
segmentation | 768 | 3 |
fpw_aff_base_ft_1024_slits_pgbmi.pth |
segmentation | 1024 | 3 |
from affmae import AFFMAE
from affmae.data.weights import EMWeights
model = AFFMAE.from_checkpoint(EMWeights.AFFMAE_BASE_FT_512)
Licence follows the source repository, which is pending lab review.