HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Add MobileSAM auto annotation feature 🚀 (#18374)

Browse source 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Muhammad Rizwan Munawar 2024-12-25 17:30:12 +05:00 • committed by GitHub
parent d88e923595
commit 93862d3640
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 30 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

16
docs/en/models/mobile-sam.md
View file

@ -130,6 +130,22 @@ You can download the model [here](https://github.com/ChaoningZhang/MobileSAM/blo
We have implemented `MobileSAM` and `SAM` using the same API. For more usage information, please see the [SAM page](sam.md).
### Automatically Build Segmentation Datasets Leveraging a Detection Model
To automatically annotate your dataset using the Ultralytics framework, utilize the `auto_annotate` function as demonstrated below:
!!! example
=== "Python"
```python
from ultralytics.data.annotator import auto_annotate
auto_annotate(data="path/to/images", det_model="yolo11x.pt", sam_model="mobile_sam.pt")
```
{% include "macros/sam-auto-annotate.md" %}
## Citations and Acknowledgements
If you find MobileSAM useful in your research or development work, please consider citing our paper: