Hugging Face's logo

kornia
/
mobile_sam

Image Segmentation
kornia
Model card Files
xet
 Community
mobile_sam
File size: 1,016 Bytes
33b7e48
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
---
license: apache-2.0
tags:
  - kornia
  - image-segmentation
---

# kornia/mobile_sam

Pretrained weights for **MobileSAM**,
used by [`kornia.models.SegmentAnything`](https://kornia.readthedocs.io/en/latest/models.html).

MobileSAM replaces the heavy ViT-H image encoder of SAM with a distilled TinyViT-5M
encoder, reducing model size by 60× while retaining most of SAM's segmentation quality.
arXiv 2023.

**Original repo:** [ChaoningZhang/MobileSAM](https://github.com/ChaoningZhang/MobileSAM)

## Weights

| File | Description |
|------|-------------|
| `mobile_sam.pt` | TinyViT-5M encoder + SAM decoder |

## Citation

```bibtex
@article{zhang2023mobilesam,
    title   = {Faster Segment Anything: Towards Lightweight SAM
               for Mobile Applications},
    author  = {Zhang, Chaoning and Han, Dongshen and Qiao, Yu
               and Kim, Jung Uk and Bae, Sung-Ho and Lee, Seungkyu
               and Hong, Choong Seon},
    journal = {arXiv preprint arXiv:2306.14289},
    year    = {2023}
}
```