Saltar al contenido

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
class RTDETR(Model):
    """
    Interface for Baidu's RT-DETR model. This Vision Transformer-based object detector provides real-time performance
    with high accuracy. It supports efficient hybrid encoding, IoU-aware query selection, and adaptable inference speed.

    Attributes:
        model (str): Path to the pre-trained model. Defaults to 'rtdetr-l.pt'.
    """

    def __init__(self, model="rtdetr-l.pt") -> None:
        """
        Initializes the RT-DETR model with the given pre-trained model file. Supports .pt and .yaml formats.

        Args:
            model (str): Path to the pre-trained model. Defaults to 'rtdetr-l.pt'.

        Raises:
            NotImplementedError: If the model file extension is not 'pt', 'yaml', or 'yml'.
        """
        super().__init__(model=model, task="detect")

    @property
    def task_map(self) -> dict:
        """
        Returns a task map for RT-DETR, associating tasks with corresponding Ultralytics classes.

        Returns:
            dict: A dictionary mapping task names to Ultralytics task classes for the RT-DETR model.
        """
        return {
            "detect": {
                "predictor": RTDETRPredictor,
                "validator": RTDETRValidator,
                "trainer": RTDETRTrainer,
                "model": RTDETRDetectionModel,
            }
        }

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'.

C贸digo fuente en ultralytics/models/rtdetr/model.py
def __init__(self, model="rtdetr-l.pt") -> None:
    """
    Initializes the RT-DETR model with the given pre-trained model file. Supports .pt and .yaml formats.

    Args:
        model (str): Path to the pre-trained model. Defaults to 'rtdetr-l.pt'.

    Raises:
        NotImplementedError: If the model file extension is not 'pt', 'yaml', or 'yml'.
    """
    super().__init__(model=model, task="detect")





Created 2023-11-12, Updated 2024-06-02
Authors: glenn-jocher (5), Burhan-Q (1)