xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-24 06:14:55 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
yolov10/docs/models/sam.md
Glenn Jocher d1107ca4cb
ultralytics 8.0.97 confusion matrix, windows, docs updates (#2511) 
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>
2023-05-09 21:20:34 +02:00

919 B
Raw Blame History

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 ❌