metadata
license: mit
tags:
- medical-imaging
- segmentation
- retinal
- pytorch
library_name: pytorch
inference: true
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