Phân khúc di động Bất cứ thứ gì ( MobileSAM )
MobileSAM is a compact, efficient image segmentation model purpose-built for mobile and edge devices. Designed to bring the power of Meta's Segment Anything Model (SAM) to environments with limited compute, MobileSAM delivers near-instant segmentation while maintaining compatibility with the original SAM pipeline. Whether you're developing real-time applications or lightweight deployments, MobileSAM provides impressive segmentation results with a fraction of the size and speed requirements of its predecessors.
Đồng hồ: Làm thế nào để chạy suy luận với MobileSAM sử dụng Ultralytics | Hướng dẫn từng bước 🎉
MobileSAM has been adopted in a variety of projects, including Grounding-SAM, AnyLabeling, and Segment Anything in 3D.
MobileSAM was trained on a single GPU using a 100k image dataset (1% of the original images) in less than a day. The training code will be released in the future.
Các mô hình có sẵn, nhiệm vụ được hỗ trợ và chế độ hoạt động
The table below outlines the available MobileSAM model, its pre-trained weights, supported tasks, and compatibility with different operating modes such as Inference, Validation, Training, and Export. Supported modes are indicated by ✅ and unsupported modes by ❌.
Kiểu mẫu | Tạ đã được tập luyện trước | Nhiệm vụ được hỗ trợ | Suy luận | Xác thực | Đào tạo | Xuất khẩu |
---|---|---|---|---|---|---|
MobileSAM | di động_sam.pt | Phân đoạn trường hợp | ✅ | ❌ | ❌ | ❌ |
MobileSAM So sánh với YOLO
The following comparison highlights the differences between Meta's SAM variants, MobileSAM, and Ultralytics' smallest segmentation models, including YOLO11n-seg:
Người mẫu | Kích cỡ (MB) |
Các tham số (Nam) |
Tốc độ ( CPU ) (cô/anh) |
---|---|---|---|
Siêu dữ liệu SAM -b | 375 | 93.7 | 49401 |
Siêu dữ liệu SAM2 -b | 162 | 80.8 | 31901 |
Siêu dữ liệu SAM2 -t | 78.1 | 38.9 | 25997 |
MobileSAM | 40.7 | 10.1 | 25381 |
FastSAM -s với YOLOv8 xương sống | 23.7 | 11.8 | 55.9 |
Ultralytics YOLOv8n -phân đoạn | 6,7 (nhỏ hơn 11,7 lần) | 3.4 (giảm 11,4 lần) | 24,5 (nhanh hơn 1061 lần) |
Ultralytics YOLO11n-phân đoạn | 5.9 (nhỏ hơn 13,2 lần) | 2,9 (giảm 13,4 lần) | 30.1 (nhanh hơn 864 lần) |
This comparison demonstrates the substantial differences in model size and speed between SAM variants and YOLO segmentation models. While SAM models offer unique automatic segmentation capabilities, YOLO models—especially YOLOv8n-seg and YOLO11n-seg—are significantly smaller, faster, and more computationally efficient.
Tests were conducted on a 2025 Apple M4 Pro with 24GB RAM using torch==2.6.0
Và ultralytics==8.3.90
. To reproduce these results:
Ví dụ
from ultralytics import ASSETS, SAM, YOLO, FastSAM
# Profile SAM2-t, SAM2-b, SAM-b, MobileSAM
for file in ["sam_b.pt", "sam2_b.pt", "sam2_t.pt", "mobile_sam.pt"]:
model = SAM(file)
model.info()
model(ASSETS)
# Profile FastSAM-s
model = FastSAM("FastSAM-s.pt")
model.info()
model(ASSETS)
# Profile YOLO models
for file_name in ["yolov8n-seg.pt", "yolo11n-seg.pt"]:
model = YOLO(file_name)
model.info()
model(ASSETS)
Thích nghi từ SAM ĐẾN MobileSAM
MobileSAM retains the same pipeline as the original SAM, including pre-processing, post-processing, and all interfaces. This means you can transition from SAM to MobileSAM with minimal changes to your workflow.
The key difference is the image encoder: MobileSAM replaces the original ViT-H encoder (632M parameters) with a much smaller Tiny-ViT encoder (5M parameters). On a single GPU, MobileSAM processes an image in about 12ms (8ms for the encoder, 4ms for the mask decoder).
ViT-Based Image Encoder Comparison
Bộ mã hóa hình ảnh | Nguyên bản SAM | MobileSAM |
---|---|---|
Các tham số | 611M | 5M |
Tốc độ | 452ms | 8ms |
Prompt-Guided Mask Decoder
Bộ giải mã mặt nạ | Nguyên bản SAM | MobileSAM |
---|---|---|
Các tham số | 3.876M | 3.876M |
Tốc độ | 4ms | 4ms |
Whole Pipeline Comparison
Toàn bộ đường ống (Enc+Dec) | Nguyên bản SAM | MobileSAM |
---|---|---|
Các tham số | 615M | 9.66M |
Tốc độ | 456ms | 12ms |
The performance of MobileSAM and the original SAM is illustrated below using both point and box prompts.
MobileSAM is approximately 5 times smaller and 7 times faster than FastSAM. For further details, visit the MobileSAM project page.
Kiểm tra MobileSAM TRONG Ultralytics
Just like the original SAM, Ultralytics provides a simple interface for testing MobileSAM, supporting both Point and Box prompts.
Tải xuống mô hình
Download the MobileSAM pretrained weights from Ultralytics assets.
Điểm nhắc nhở
Ví dụ
from ultralytics import SAM
# Load the model
model = SAM("mobile_sam.pt")
# Predict a segment based on a single point prompt
model.predict("ultralytics/assets/zidane.jpg", points=[900, 370], labels=[1])
# Predict multiple segments based on multiple points prompt
model.predict("ultralytics/assets/zidane.jpg", points=[[400, 370], [900, 370]], labels=[1, 1])
# Predict a segment based on multiple points prompt per object
model.predict("ultralytics/assets/zidane.jpg", points=[[[400, 370], [900, 370]]], labels=[[1, 1]])
# Predict a segment using both positive and negative prompts.
model.predict("ultralytics/assets/zidane.jpg", points=[[[400, 370], [900, 370]]], labels=[[1, 0]])
Hộp nhắc nhở
Ví dụ
from ultralytics import SAM
# Load the model
model = SAM("mobile_sam.pt")
# Predict a segment based on a single point prompt
model.predict("ultralytics/assets/zidane.jpg", points=[900, 370], labels=[1])
# Predict multiple segments based on multiple points prompt
model.predict("ultralytics/assets/zidane.jpg", points=[[400, 370], [900, 370]], labels=[1, 1])
# Predict a segment based on multiple points prompt per object
model.predict("ultralytics/assets/zidane.jpg", points=[[[400, 370], [900, 370]]], labels=[[1, 1]])
# Predict a segment using both positive and negative prompts.
model.predict("ultralytics/assets/zidane.jpg", points=[[[400, 370], [900, 370]]], labels=[[1, 0]])
Both MobileSAM
Và SAM
share the same API. For more usage details, see the SAM documentation.
Automatically Build Segmentation Datasets Using a Detection Model
To automatically annotate your dataset with the Ultralytics framework, use the auto_annotate
chức năng như hiển thị bên dưới:
Ví dụ
Lý lẽ | Kiểu | Mặc định | Sự miêu tả |
---|---|---|---|
data |
str |
yêu cầu | Đường dẫn đến thư mục chứa hình ảnh mục tiêu để chú thích hoặc phân đoạn. |
det_model |
str |
'yolo11x.pt' |
YOLO đường dẫn mô hình phát hiện để phát hiện đối tượng ban đầu. |
sam_model |
str |
'sam_b.pt' |
SAM đường dẫn mô hình cho phân đoạn (hỗ trợ SAM , SAM2 các biến thể và mô hình mobile_sam). |
device |
str |
'' |
Thiết bị tính toán (ví dụ, ' cuda :0', ' cpu ' hoặc '' để tự động phát hiện thiết bị). |
conf |
float |
0.25 |
YOLO ngưỡng tin cậy phát hiện để lọc các phát hiện yếu. |
iou |
float |
0.45 |
Ngưỡng IoU cho việc loại bỏ không tối đa để lọc các hộp chồng chéo. |
imgsz |
int |
640 |
Kích thước đầu vào để thay đổi kích thước hình ảnh (phải là bội số của 32). |
max_det |
int |
300 |
Số lượng phát hiện tối đa trên mỗi hình ảnh để tăng hiệu quả bộ nhớ. |
classes |
list[int] |
None |
Danh sách các chỉ số lớp để phát hiện (ví dụ: [0, 1] dành cho người và xe đạp). |
output_dir |
str |
None |
Lưu thư mục để chú thích (mặc định là './labels' theo đường dẫn dữ liệu). |
Trích dẫn và Lời cảm ơn
If MobileSAM is helpful in your research or development, please consider citing the following paper:
Read the full MobileSAM paper on arXiv.
CÂU HỎI THƯỜNG GẶP
What Is MobileSAM and How Does It Differ from the Original SAM Model?
MobileSAM is a lightweight, fast image segmentation model optimized for mobile and edge applications. It maintains the same pipeline as the original SAM but replaces the large ViT-H encoder (632M parameters) with a compact Tiny-ViT encoder (5M parameters). This results in MobileSAM being about 5 times smaller and 7 times faster than the original SAM, operating at roughly 12ms per image versus SAM's 456ms. Explore more about MobileSAM's implementation on the MobileSAM GitHub repository.
How Can I Test MobileSAM Using Ultralytics?
Testing MobileSAM in Ultralytics is straightforward. You can use Point and Box prompts to predict segments. For example, using a Point prompt:
from ultralytics import SAM
# Load the model
model = SAM("mobile_sam.pt")
# Predict a segment based on a point prompt
model.predict("ultralytics/assets/zidane.jpg", points=[900, 370], labels=[1])
For more details, see the Testing MobileSAM in Ultralytics section.
Why Should I Use MobileSAM for My Mobile Application?
MobileSAM is ideal for mobile and edge applications due to its lightweight design and rapid inference speed. Compared to the original SAM, MobileSAM is about 5 times smaller and 7 times faster, making it suitable for real-time segmentation on devices with limited computational resources. Its efficiency enables mobile devices to perform real-time image segmentation without significant latency. Additionally, MobileSAM supports Inference mode optimized for mobile performance.
How Was MobileSAM Trained, and Is the Training Code Available?
MobileSAM was trained on a single GPU with a 100k image dataset (1% of the original images) in under a day. While the training code will be released in the future, you can currently access pre-trained weights and implementation details from the MobileSAM GitHub repository.
What Are the Primary Use Cases for MobileSAM?
MobileSAM is designed for fast, efficient image segmentation in mobile and edge environments. Primary use cases include:
- Real-time object detection and segmentation for mobile apps
- Low-latency image processing on devices with limited compute
- Integration in AI-powered mobile applications for augmented reality (AR), analytics, and more
For more details on use cases and performance, see Adapting from SAM to MobileSAM and the Ultralytics blog on MobileSAM applications.