You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
37 lines
786 B
37 lines
786 B
2 years ago
|
---
|
||
|
|
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: |
|