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.
976 B
976 B
Install
pip install ultralytics
Development
git clone https://github.com/ultralytics/ultralytics
cd ultralytics
pip install -e .
Usage
1. CLI
To simply use the latest Ultralytics YOLO models
yolo task=detect mode=train model=s.yaml ...
classify infer s-cls.yaml
segment val s-seg.yaml
2. Python SDK
To use pythonic interface of Ultralytics YOLO model
import ultralytics
from ultralytics import YOLO
model = YOLO()
model.new("s-seg.yaml") # automatically detects task type
model.load("s-seg.pt") # load checkpoint
model.train(data="coco128-segments", epochs=1, lr0=0.01, ...)
If you're looking to modify YOLO for R&D or to build on top of it, refer to Using Trainer Guide on our docs.