Référence pour ultralytics/models/rtdetr/model.py
Note
Ce fichier est disponible à l'adresse https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/models/rtdetr/model .py. Si tu repères un problème, aide à le corriger en contribuant à une Pull Request 🛠️. Merci 🙏 !
ultralytics.models.rtdetr.model.RTDETR
Bases : Model
Interface pour le modèle RT-DETR de Baidu. Ce détecteur d'objets basé sur Vision Transformer offre une performance en temps réel avec une grande précision. Il prend en charge un codage hybride efficace, une sélection de requêtes tenant compte de l'expérience utilisateur et une vitesse d'inférence adaptable.
Attributs :
Nom | Type | Description |
---|---|---|
model |
str
|
Chemin d'accès au modèle pré-entraîné. La valeur par défaut est 'rtdetr-l.pt'. |
Code source dans ultralytics/models/rtdetr/model.py
task_map: dict
property
Renvoie une carte des tâches pour RT-DETR, en associant les tâches aux classes Ultralytics correspondantes.
Retourne :
Nom | Type | Description |
---|---|---|
dict |
dict
|
Un dictionnaire qui fait correspondre les noms des tâches aux classes de tâches de Ultralytics pour le modèle RT-DETR . |
__init__(model='rtdetr-l.pt')
Initialise le modèle RT-DETR avec le fichier de modèle pré-entraîné donné. Prend en charge les formats .pt et .yaml.
Paramètres :
Nom | Type | Description | DĂ©faut |
---|---|---|---|
model |
str
|
Chemin d'accès au modèle pré-entraîné. La valeur par défaut est 'rtdetr-l.pt'. |
'rtdetr-l.pt'
|
Augmente :
Type | Description |
---|---|
NotImplementedError
|
Si l'extension du fichier de modèle n'est pas 'pt', 'yaml' ou 'yml'. |