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.
919 B
919 B
comments | description |
---|---|
true | 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:
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 | ✔️ |
Validation | ❌ |
Training | ❌ |