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.

786 B

comments
true

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(...)  # 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 ✔️
Validation
Training