a yf@sTdZddlmZddlmZddlmZddlmZddl m Z Gdd d eZ d S) a~ Interface for Baidu's RT-DETR, a Vision Transformer-based real-time object detector. RT-DETR offers real-time performance and high accuracy, excelling in accelerated backends like CUDA with TensorRT. It features an efficient hybrid encoder and IoU-aware query selection for enhanced detection accuracy. For more information on RT-DETR, visit: https://arxiv.org/pdf/2304.08069.pdf )Model)RTDETRDetectionModel)RTDETRPredictor) RTDETRTrainer)RTDETRValidatorcs:eZdZdZd ddfdd ZeedddZZS) RTDETRaT 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'. rtdetr-l.ptN)returncstj|dddS)aE 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'. detect)modeltaskN)super__init__)selfr  __class__[/var/www/html/django/DPS/env/lib/python3.9/site-packages/ultralytics/models/rtdetr/model.pyrs zRTDETR.__init__cCsdttttdiS)z 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. r )Z predictor validatorZtrainerr )rrrr)rrrrtask_map's zRTDETR.task_map)r ) __name__ __module__ __qualname____doc__rpropertydictr __classcell__rrrrrs rN) rZultralytics.engine.modelrZultralytics.nn.tasksrZpredictrtrainrvalrrrrrrs