model update

README.md

@@ -1,3 +1,45 @@
 ---
 license: mit
+tags:
+  - medical-imaging
+  - segmentation
+  - retinal
+  - pytorch
+library_name: pytorch
+inference: true
 ---
+
+# RetSAM
+
+<p align="left">
+  <a href="TBD"><img src="https://img.shields.io/badge/arXiv-PDF-red" alt="arXiv"></a>
+  <a href="https://wzhjerry.github.io/RetSAM/"><img src="https://img.shields.io/badge/Project-Page-blue" alt="Project Page"></a>
+  <a href="https://huggingface.co/JerryWzh/RetSAM_public"><img src="https://img.shields.io/badge/Model-RetSAM-green" alt="Model"></a>
+</p>
+
+
+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`.
+
+Citation
+--------
+```bibtex
+ToDO
+```
+
+Contact
+-------
+For questions or access to full RetSAM models, email: zhonghua.wang@monash.edu

retsam_v1_for_public.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fa969c7391833a2942ec82946305e7701158b6d724781312d45387007b4e6782
+size 9196220962