Referência para ultralytics/models/rtdetr/model.py
Nota
Este ficheiro está disponível em https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/models/rtdetr/model .py. Se detectares um problema, por favor ajuda a corrigi-lo contribuindo com um Pull Request 🛠️. Obrigado 🙏!
ultralytics.models.rtdetr.model.RTDETR
Bases: Model
Interface para o modelo RT-DETR da Baidu. Este detetor de objectos baseado no Vision Transformer proporciona um desempenho em tempo real com elevada precisão. Suporta uma codificação híbrida eficiente, uma seleção de consulta consciente da IoU e uma velocidade de inferência adaptável.
Atributos:
Nome | Tipo | Descrição |
---|---|---|
model |
str
|
Caminho para o modelo pré-treinado. Usa como predefinição 'rtdetr-l.pt'. |
Código fonte em ultralytics/models/rtdetr/model.py
task_map: dict
property
Devolve um mapa de tarefas para RT-DETR, associando as tarefas às classes Ultralytics correspondentes.
Devolve:
Nome | Tipo | Descrição |
---|---|---|
dict |
dict
|
Um dicionário que mapeia os nomes das tarefas para as classes de tarefas Ultralytics para o modelo RT-DETR . |
__init__(model='rtdetr-l.pt')
Inicializa o modelo RT-DETR com o ficheiro de modelo pré-treinado fornecido. Suporta os formatos .pt e .yaml.
Parâmetros:
Nome | Tipo | Descrição | Predefinição |
---|---|---|---|
model |
str
|
Caminho para o modelo pré-treinado. Usa como predefinição 'rtdetr-l.pt'. |
'rtdetr-l.pt'
|
Aumenta:
Tipo | Descrição |
---|---|
NotImplementedError
|
Se a extensão do ficheiro do modelo não for 'pt', 'yaml' ou 'yml'. |