ducha-aiki commited on
Commit
33b7e48
·
verified ·
1 Parent(s): bbcee74

Update README

Browse files
Files changed (1) hide show
  1. README.md +37 -0
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
+ ```