JerryWzh/RetSAM_public

RetSAM

Overview

Official repository for "A General Model for Retinal Segmentation and Quantification".

Note: Publicly released models are trained on public datasets only and may differ from the paper-reported results.

Checkpoint

• --out_channels: "(2,3,2,4,6)".
• --patch_size: 4.
• --window_size: 10.
• --depths: "(2, 2, 18, 2)".
• --num_heads: "(4, 8, 16, 32)".
• --feature_size: 128.
• --size: 640.

Download from Hugging Face

Install huggingface_hub:

pip install -U huggingface_hub

Download the checkpoint with Python:

from huggingface_hub import hf_hub_download

ckpt_path = hf_hub_download(
    repo_id="JerryWzh/RetSAM_public",
    filename="retsam_v1_for_public.ckpt",
)
print(ckpt_path)

Or download with CLI:

hf download JerryWzh/RetSAM_public retsam_v1_for_public.ckpt

Citation

@article{wang2026general,
  title={A General Model for Retinal Segmentation and Quantification},
  author={Wang, Zhonghua and Ju, Lie and Li, Sijia and Feng, Wei and Zhou, Sijin and Hu, Ming and Xiong, Jianhao and Tang, Xiaoying and Peng, Yifan and Lin, Mingquan and others},
  journal={arXiv preprint arXiv:2602.07012},
  year={2026}
}

Contact

For questions or access to full RetSAM models, please contact me at: zhonghua.wang@monash.edu