Reference for ultralytics/models/yolo/model.py
Note
This file is available at https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/model.py. If you spot a problem please help fix it by contributing a Pull Request 🛠️. Thank you 🙏!
ultralytics.models.yolo.model.YOLO
Bases: Model
YOLO (You Only Look Once) object detection model.
Source code in ultralytics/models/yolo/model.py
task_map
property
Map head to model, trainer, validator, and predictor classes.
__init__(model='yolov8n.pt', task=None, verbose=False)
Initialize YOLO model, switching to YOLOWorld if model filename contains '-world'.
Source code in ultralytics/models/yolo/model.py
ultralytics.models.yolo.model.YOLOWorld
Bases: Model
YOLO-World object detection model.
Source code in ultralytics/models/yolo/model.py
task_map
property
Map head to model, validator, and predictor classes.
__init__(model='yolov8s-world.pt', verbose=False)
Initializes the YOLOv8-World model with the given pre-trained model file. Supports .pt and .yaml formats.
Parameters:
Name | Type | Description | Default |
---|---|---|---|
model |
str | Path
|
Path to the pre-trained model. Defaults to 'yolov8s-world.pt'. |
'yolov8s-world.pt'
|
Source code in ultralytics/models/yolo/model.py
set_classes(classes)
Set classes.
Parameters:
Name | Type | Description | Default |
---|---|---|---|
classes |
List(str
|
A list of categories i.e ["person"]. |
required |