Ссылка для ultralytics/models/rtdetr/model.py
Примечание
Этот файл доступен по адресу https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/models/rtdetr/model .py. Если ты заметил проблему, пожалуйста, помоги исправить ее, отправив Pull Request 🛠️. Спасибо 🙏!
ultralytics.models.rtdetr.model.RTDETR
Базы: Model
Интерфейс для модели Baidu RT-DETR . Этот детектор объектов на основе Vision Transformer обеспечивает производительность в реальном времени с высокой точностью. Он поддерживает эффективное гибридное кодирование, выбор запросов с учетом IoU и адаптируемую скорость вывода.
Атрибуты:
Имя | Тип | Описание |
---|---|---|
model |
str
|
Путь к предварительно обученной модели. По умолчанию - 'rtdetr-l.pt'. |
Исходный код в ultralytics/models/rtdetr/model.py
task_map: dict
property
Возвращает карту задач для RT-DETR, связывая задачи с соответствующими классами Ultralytics .
Возвращается:
Имя | Тип | Описание |
---|---|---|
dict |
dict
|
Словарь, отображающий названия задач на классы задач Ultralytics для модели RT-DETR . |
__init__(model='rtdetr-l.pt')
Инициализирует модель RT-DETR с заданным файлом предварительно обученной модели. Поддерживает форматы .pt и .yaml.
Параметры:
Имя | Тип | Описание | По умолчанию |
---|---|---|---|
model |
str
|
Путь к предварительно обученной модели. По умолчанию - 'rtdetr-l.pt'. |
'rtdetr-l.pt'
|
Поднимает:
Тип | Описание |
---|---|
NotImplementedError
|
Если расширение файла модели не 'pt', 'yaml' или 'yml'. |