Riferimento per ultralytics/models/rtdetr/model.py
Nota
Questo file è disponibile su https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/models/rtdetr/model .py. Se riscontri un problema, contribuisci a risolverlo inviando una Pull Request 🛠️. Grazie 🙏!
ultralytics.models.rtdetr.model.RTDETR
Basi: Model
Interfaccia per il modello RT-DETR di Baidu. Questo rilevatore di oggetti basato su Vision Transformer offre prestazioni in tempo reale con un'elevata precisione. Supporta una codifica ibrida efficiente, una selezione delle query consapevole dell'IoU e una velocità di inferenza adattabile.
Attributi:
Nome | Tipo | Descrizione |
---|---|---|
model |
str
|
Percorso del modello pre-addestrato. Il valore predefinito è 'rtdetr-l.pt'. |
Codice sorgente in ultralytics/models/rtdetr/model.py
task_map: dict
property
Restituisce una mappa dei compiti per RT-DETR, associando i compiti alle classi corrispondenti di Ultralytics .
Restituzione:
Nome | Tipo | Descrizione |
---|---|---|
dict |
dict
|
Un dizionario che mappa i nomi dei compiti alle classi di compiti di Ultralytics per il modello RT-DETR . |
__init__(model='rtdetr-l.pt')
Inizializza il modello RT-DETR con il file del modello pre-addestrato indicato. Supporta i formati .pt e .yaml.
Parametri:
Nome | Tipo | Descrizione | Predefinito |
---|---|---|---|
model |
str
|
Percorso del modello pre-addestrato. Il valore predefinito è 'rtdetr-l.pt'. |
'rtdetr-l.pt'
|
Aumenta:
Tipo | Descrizione |
---|---|
NotImplementedError
|
Se l'estensione del file del modello non è 'pt', 'yaml' o 'yml'. |