Referentie voor ultralytics/models/rtdetr/model.py
Opmerking
Dit bestand is beschikbaar op https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/models/rtdetr/model .py. Als je een probleem ziet, help het dan oplossen door een Pull Request 🛠️ bij te dragen. Bedankt 🙏!
ultralytics.models.rtdetr.model.RTDETR
Basis: Model
Interface voor Baidu's RT-DETR model. Deze op Vision Transformer gebaseerde objectdetector levert realtime prestaties met hoge nauwkeurigheid. Het ondersteunt efficiënte hybride codering, IoU-bewuste vraagselectie en aanpasbare inferentiesnelheid.
Kenmerken:
Naam | Type | Beschrijving |
---|---|---|
model |
str
|
Pad naar het voorgetrainde model. Wordt standaard ingesteld op 'rtdetr-l.pt'. |
Broncode in ultralytics/models/rtdetr/model.py
task_map: dict
property
Geeft een takenmap terug voor RT-DETR, die taken koppelt aan de bijbehorende klassen van Ultralytics .
Retourneert:
Naam | Type | Beschrijving |
---|---|---|
dict |
dict
|
Een woordenboek waarin taaknamen worden toegewezen aan Ultralytics taakklassen voor het RT-DETR model. |
__init__(model='rtdetr-l.pt')
Initialiseert het RT-DETR model met het gegeven voorgetrainde modelbestand. Ondersteunt .pt en .yaml formaten.
Parameters:
Naam | Type | Beschrijving | Standaard |
---|---|---|---|
model |
str
|
Pad naar het voorgetrainde model. Wordt standaard ingesteld op 'rtdetr-l.pt'. |
'rtdetr-l.pt'
|
Verhogingen:
Type | Beschrijving |
---|---|
NotImplementedError
|
Als de extensie van het modelbestand niet 'pt', 'yaml' of 'yml' is. |