|hZdZddlmZddlmZddlmZddlmZddl m Z Gdd eZ y ) ag 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. References: https://arxiv.org/pdf/2304.08069.pdf )Model)RTDETRDetectionModel)RTDETRPredictor) RTDETRTrainer)RTDETRValidatorcDeZdZdZddeddffd ZedefdZxZ S)RTDETRa Interface for Baidu's RT-DETR model, a Vision Transformer-based real-time object detector. This model 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. Methods: task_map: Return a task map for RT-DETR, associating tasks with corresponding Ultralytics classes. Examples: Initialize RT-DETR with a pre-trained model >>> from ultralytics import RTDETR >>> model = RTDETR("rtdetr-l.pt") >>> results = model("image.jpg") modelreturnNc(t||dy)z Initialize the RT-DETR model with the given pre-trained model file. Args: model (str): Path to the pre-trained model. Supports .pt, .yaml, and .yml formats. detect)r taskN)super__init__)selfr __class__s ^/var/www/html/test/engine/venv/lib/python3.12/site-packages/ultralytics/models/rtdetr/model.pyrzRTDETR.__init__(s u84c4dttttdiS)z Return 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) predictor validatortrainerr )rrrr)rs rtask_mapzRTDETR.task_map1s" ,,(-   r)z rtdetr-l.pt) __name__ __module__ __qualname____doc__strrpropertydictr __classcell__)rs@rr r s6&5c5d5 $  rr N) rultralytics.engine.modelrultralytics.nn.tasksrpredictrtrainrvalrr rrr)s'+5$ , U, r