Update README
Browse files
README.md
ADDED
|
@@ -0,0 +1,37 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
tags:
|
| 4 |
+
- kornia
|
| 5 |
+
- image-segmentation
|
| 6 |
+
---
|
| 7 |
+
|
| 8 |
+
# kornia/mobile_sam
|
| 9 |
+
|
| 10 |
+
Pretrained weights for **MobileSAM**,
|
| 11 |
+
used by [`kornia.models.SegmentAnything`](https://kornia.readthedocs.io/en/latest/models.html).
|
| 12 |
+
|
| 13 |
+
MobileSAM replaces the heavy ViT-H image encoder of SAM with a distilled TinyViT-5M
|
| 14 |
+
encoder, reducing model size by 60× while retaining most of SAM's segmentation quality.
|
| 15 |
+
arXiv 2023.
|
| 16 |
+
|
| 17 |
+
**Original repo:** [ChaoningZhang/MobileSAM](https://github.com/ChaoningZhang/MobileSAM)
|
| 18 |
+
|
| 19 |
+
## Weights
|
| 20 |
+
|
| 21 |
+
| File | Description |
|
| 22 |
+
|------|-------------|
|
| 23 |
+
| `mobile_sam.pt` | TinyViT-5M encoder + SAM decoder |
|
| 24 |
+
|
| 25 |
+
## Citation
|
| 26 |
+
|
| 27 |
+
```bibtex
|
| 28 |
+
@article{zhang2023mobilesam,
|
| 29 |
+
title = {Faster Segment Anything: Towards Lightweight SAM
|
| 30 |
+
for Mobile Applications},
|
| 31 |
+
author = {Zhang, Chaoning and Han, Dongshen and Qiao, Yu
|
| 32 |
+
and Kim, Jung Uk and Bae, Sung-Ho and Lee, Seungkyu
|
| 33 |
+
and Hong, Choong Seon},
|
| 34 |
+
journal = {arXiv preprint arXiv:2306.14289},
|
| 35 |
+
year = {2023}
|
| 36 |
+
}
|
| 37 |
+
```
|