Referencia para ultralytics/models/rtdetr/model.py
Nota
Este archivo está disponible en https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/models/rtdetr/model .py. Si detectas algún problema, por favor, ayuda a solucionarlo contribuyendo con una Pull Request 🛠️. ¡Gracias 🙏!
ultralytics.models.rtdetr.model.RTDETR
Bases: Model
Interfaz para el modelo RT-DETR de Baidu. Este detector de objetos basado en el Transformador de Visión proporciona un rendimiento en tiempo real con gran precisión. Admite una codificación híbrida eficiente, una selección de consultas consciente de la IoU y una velocidad de inferencia adaptable.
Atributos:
Nombre | Tipo | Descripción |
---|---|---|
model |
str
|
Ruta al modelo preentrenado. Por defecto es 'rtdetr-l.pt'. |
Código fuente en ultralytics/models/rtdetr/model.py
task_map: dict
property
Devuelve un mapa de tareas para RT-DETR, asociando las tareas a las clases Ultralytics correspondientes.
Devuelve:
Nombre | Tipo | Descripción |
---|---|---|
dict |
dict
|
Un diccionario que asigna nombres de tareas a clases de tareas de Ultralytics para el modelo RT-DETR . |
__init__(model='rtdetr-l.pt')
Inicializa el modelo RT-DETR con el archivo de modelo preentrenado dado. Admite los formatos .pt y .yaml.
Parámetros:
Nombre | Tipo | Descripción | Por defecto |
---|---|---|---|
model |
str
|
Ruta al modelo preentrenado. Por defecto es 'rtdetr-l.pt'. |
'rtdetr-l.pt'
|
Aumenta:
Tipo | Descripción |
---|---|
NotImplementedError
|
Si la extensión del archivo del modelo no es 'pt', 'yaml' o 'yml'. |