mirror of
https://github.com/THU-MIG/yolov10.git
synced 2025-05-23 21:44:22 +08:00
3fd317edfd
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
786 B
Markdown
37 lines
786 B
Markdown
---
|
|
comments: true
|
|
---
|
|
|
|
# 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(...) # train the model
|
|
```
|
|
|
|
# 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: |
|