§ —/jÝãón—dZddlmZddlmZddlmZddlmZddl m Z ddl m Z Gd „d e¦«Z d S) 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)Ú TORCH_1_11é)ÚRTDETRPredictor)Ú RTDETRTrainer)ÚRTDETRValidatorcóJ‡—eZdZdZddeddfˆfd„ Zedefd„¦«ZˆxZ 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") ú rtdetr-l.ptÚmodelÚreturnNcóp•—ts Jd¦«‚t¦« |d¬¦«dS)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. zRTDETR requires torch>=1.11Údetect)r ÚtaskN)rÚsuperÚ__init__)Úselfr Ú __class__s €úd/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/ultralytics/models/rtdetr/model.pyrzRTDETR.__init__(s<ø€õ Ð8Ð8Ð8Ñ8Ô8ˆzÝ ‰Œ×Ò˜u¨8ÐÑ4Ô4Ð4Ð4Ð4ócó<—dttttdœiS)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Ú validatorÚtrainerr )rr rr)rs rÚtask_mapzRTDETR.task_map1s'€ð Ý,Ý,Ý(Ý-ð ðð ð r)r ) Ú__name__Ú __module__Ú __qualname__Ú__doc__ÚstrrÚpropertyÚdictrÚ __classcell__)rs@rr r sø€€€€€ððð$5ð5˜cð5°dð5ð5ð5ð5ð5ð5ðð  ˜$ð  ð  ð  ñ„Xð  ð  ð  ð  ð  rr N)r Úultralytics.engine.modelrÚultralytics.nn.tasksrÚultralytics.utils.torch_utilsrÚpredictrÚtrainrÚvalr r ©rrúr,s±ðððð+Ð*Ð*Ð*Ð*Ð*Ø5Ð5Ð5Ð5Ð5Ð5Ø4Ð4Ð4Ð4Ð4Ð4à$Ð$Ð$Ð$Ð$Ð$Ø Ð Ð Ð Ð Ð Ø Ð Ð Ð Ð Ð ð* ð* ð* ð* ð* ˆUñ* ô* ð* ð* ð* r