mirror of
https://github.com/THU-MIG/yolov10.git
synced 2025-05-24 14:44:21 +08:00
d1107ca4cb
Co-authored-by: Yonghye Kwon <developer.0hye@gmail.com> Co-authored-by: Dowon <ks2515@naver.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
37 lines
919 B
Markdown
37 lines
919 B
Markdown
---
|
|
comments: true
|
|
description: Learn about the Vision Transformer (ViT) and segment anything with SAM models. Train and use pre-trained models with Python API.
|
|
---
|
|
|
|
# Vision Transformers
|
|
|
|
Vit models currently support Python environment:
|
|
|
|
```python
|
|
from ultralytics.vit import SAM
|
|
|
|
# from ultralytics.vit import MODEL_TYPE
|
|
|
|
model = SAM("sam_b.pt")
|
|
model.info() # display model information
|
|
model.predict(...) # predict
|
|
```
|
|
|
|
# Segment Anything
|
|
|
|
## About
|
|
|
|
## Supported Tasks
|
|
|
|
| Model Type | Pre-trained Weights | Tasks Supported |
|
|
|------------|---------------------|-----------------------|
|
|
| sam base | `sam_b.pt` | Instance Segmentation |
|
|
| sam large | `sam_l.pt` | Instance Segmentation |
|
|
|
|
## Supported Modes
|
|
|
|
| Mode | Supported |
|
|
|------------|--------------------|
|
|
| Inference | :heavy_check_mark: |
|
|
| Validation | :x: |
|
|
| Training | :x: | |