Reference for ultralytics/models/rtdetr/model.py
Note
This file is available at https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/rtdetr/model.py. If you spot a problem please help fix it by contributing a Pull Request 🛠️. Thank you 🙏!
ultralytics.models.rtdetr.model.RTDETR
Bases: Model
Interface for Baidu's RT-DETR model. This Vision Transformer-based object detector provides real-time performance with high accuracy. It supports efficient hybrid encoding, IoU-aware query selection, and adaptable inference speed.
Attributes:
Name | Type | Description |
---|---|---|
model |
str
|
Path to the pre-trained model. Defaults to 'rtdetr-l.pt'. |
Source code in ultralytics/models/rtdetr/model.py
task_map: dict
property
Returns a task map for RT-DETR, associating tasks with corresponding Ultralytics classes.
Returns:
Name | Type | Description |
---|---|---|
dict |
dict
|
A dictionary mapping task names to Ultralytics task classes for the RT-DETR model. |
__init__(model='rtdetr-l.pt')
Initializes the RT-DETR model with the given pre-trained model file. Supports .pt and .yaml formats.
Parameters:
Name | Type | Description | Default |
---|---|---|---|
model |
str
|
Path to the pre-trained model. Defaults to 'rtdetr-l.pt'. |
'rtdetr-l.pt'
|
Raises:
Type | Description |
---|---|
NotImplementedError
|
If the model file extension is not 'pt', 'yaml', or 'yml'. |