h"ddlmZddlmZmZmZmZmZddlZddl m Z ddl m Z m Z mZddlmZddlmZmZdZGd d e ZGd d eZy) )Path)AnyDictListTupleUnionN) YOLODataset)ComposeFormat v8_transforms)DetectionValidator)colorstrops)RTDETRValidatorc<eZdZdZddfd Zdfd ZddZxZS) RTDETRDataseta Real-Time DEtection and TRacking (RT-DETR) dataset class extending the base YOLODataset class. This specialized dataset class is designed for use with the RT-DETR object detection model and is optimized for real-time detection and tracking tasks. Attributes: augment (bool): Whether to apply data augmentation. rect (bool): Whether to use rectangular training. use_segments (bool): Whether to use segmentation masks. use_keypoints (bool): Whether to use keypoint annotations. imgsz (int): Target image size for training. Methods: load_image: Load one image from dataset index. build_transforms: Build transformation pipeline for the dataset. Examples: Initialize an RT-DETR dataset >>> dataset = RTDETRDataset(img_path="path/to/images", imgsz=640) >>> image, hw = dataset.load_image(0) N)datac(t||d|i|y)aU Initialize the RTDETRDataset class by inheriting from the YOLODataset class. This constructor sets up a dataset specifically optimized for the RT-DETR (Real-Time DEtection and TRacking) model, building upon the base YOLODataset functionality. Args: *args (Any): Variable length argument list passed to the parent YOLODataset class. data (dict | None): Dictionary containing dataset information. If None, default values will be used. **kwargs (Any): Additional keyword arguments passed to the parent YOLODataset class. rN)super__init__)selfrargskwargs __class__s [/var/www/html/eduruby.in/venv/lib/python3.12/site-packages/ultralytics/models/rtdetr/val.pyrzRTDETRDataset.__init__(s $4T4V4c&t|||S)a" Load one image from dataset index 'i'. Args: i (int): Index of the image to load. rect_mode (bool, optional): Whether to use rectangular mode for batch inference. Returns: im (torch.Tensor): The loaded image. resized_hw (tuple): Height and width of the resized image with shape (2,). Examples: Load an image from the dataset >>> dataset = RTDETRDataset(img_path="path/to/images") >>> image, hw = dataset.load_image(0) )i rect_mode)r load_image)rrrrs rr zRTDETRDataset.load_image6s"w!A!;;rc |jr|jr|js |jnd|_|jr|js |jnd|_|jr|js |jnd|_t ||j |d}n tg}|jtdd|j|jd|j|j|S)z Build transformation pipeline for the dataset. Args: hyp (dict, optional): Hyperparameters for transformations. Returns: (Compose): Composition of transformation functions. gT)stretchxywh) bbox_format normalize return_maskreturn_keypoint batch_idx mask_ratio mask_overlap)augmentrectmosaicmixupcutmixr imgszr appendr use_segments use_keypointsr) overlap_mask)rhyp transformss rbuild_transformszRTDETRDataset.build_transformsIs <<'+||DII3CJ%)\\$)) CI'+||DII3CJ&tTZZdKJ!J " -- $ 2 2>> --  r)F)N)__name__ __module__ __qualname____doc__rr r7 __classcell__)rs@rrrs.$( 5<&rrceZdZdZd dZdeejeeje ejfdee e ejffdZ de e ejfde e e fddfd Zy) ra RTDETRValidator extends the DetectionValidator class to provide validation capabilities specifically tailored for the RT-DETR (Real-Time DETR) object detection model. The class allows building of an RTDETR-specific dataset for validation, applies Non-maximum suppression for post-processing, and updates evaluation metrics accordingly. Attributes: args (Namespace): Configuration arguments for validation. data (dict): Dataset configuration dictionary. Methods: build_dataset: Build an RTDETR Dataset for validation. postprocess: Apply Non-maximum suppression to prediction outputs. Examples: Initialize and run RT-DETR validation >>> from ultralytics.models.rtdetr import RTDETRValidator >>> args = dict(model="rtdetr-l.pt", data="coco8.yaml") >>> validator = RTDETRValidator(args=args) >>> validator() Notes: For further details on the attributes and methods, refer to the parent DetectionValidator class. Nc t||jj|d|jd|jjxsdt |d|j  S)a Build an RTDETR Dataset. Args: img_path (str): Path to the folder containing images. mode (str, optional): `train` mode or `val` mode, users are able to customize different augmentations for each mode. batch (int, optional): Size of batches, this is for `rect`. Returns: (RTDETRDataset): Dataset configured for RT-DETR validation. FNz: ) img_pathr0 batch_sizer+r5r,cacheprefixr)rrr0rArr)rr?modebatchs r build_datasetzRTDETRValidator.build_datasetsT))// ))//)TtfBK(  rpredsreturnct|ttfs|dg}|dj\}}}|dj d|dz fd\}}||j j z}tjd|jg|z}t|D]\}} tj| } ||jd\} } tj| | d| dgd} | | jd } | | |j j kD||<|D cgc]} | ddddf| dddf| ddd fd !c} Scc} w) a Apply Non-maximum suppression to prediction outputs. Args: preds (torch.Tensor | List | Tuple): Raw predictions from the model. If tensor, should have shape (batch_size, num_predictions, num_classes + 4) where last dimension contains bbox coords and class scores. Returns: (List[Dict[str, torch.Tensor]]): List of dictionaries for each image, each containing: - 'bboxes': Tensor of shape (N, 4) with bounding box coordinates - 'conf': Tensor of shape (N,) with confidence scores - 'cls': Tensor of shape (N,) with class indices Nr)dim)r)device).NT) descending)bboxesconfcls) isinstancelisttupleshapesplitrr0torchzerosrM enumerater xywh2xyxymaxcatargsortrQ)rrFbs_ndrPscoresoutputsrbboxscorerRpredxs r postprocesszRTDETRValidator.postprocesssE %$/DME!HNN ArqBF <$))//!;;vfmm<=B ( 6GAt==&Dr*JE399dE)$4c)nE2ND  67Dediinn45GAJ  6PWW!1QU8Qq!tWQq!tWEWWWs!$Eprednpbatchc t|dj}|jr t|n|}|dj }|dddgfxx|dd|j j z zcc<|dddgfxx|dd|j j z zcc<tj|}|d d d dfxx|d d dd fdz zcc<t|j|d j|d jD]^\}}}|jj||jt||D cgc]} t| dc} t|d d `y cc} w)a Serialize YOLO predictions to COCO json format. Args: predn (Dict[str, torch.Tensor]): Predictions dictionary containing 'bboxes', 'conf', and 'cls' keys with bounding box coordinates, confidence scores, and class predictions. pbatch (Dict[str, Any]): Batch dictionary containing 'imgsz', 'ori_shape', 'ratio_pad', and 'im_file'. im_filerP.r ori_shapeNrQrRrO)image_id category_idrdre)rstem isnumericintclonerr0r xyxy2xywhziptolistjdictr1 class_mapround) rrirjrsrqboxbscrgs r pred_to_jsonzRTDETRValidator.pred_to_jsonsYF9%&++ $ 03t9dHo##% C!QKF;/2TYY__DD C!QKF;/2TYY__DDmmC  ArrE c!QR%j1n$ 3::<v)=)=)?uATATAVW GAq! JJ   (#'>>#aΛ4QU1a[4"1a[    5sE8)valN)r8r9r:r;rErrXTensorrrrstrrhrrrrrris4 2X5<<ell);U5<<=PPQX d3 $% &XB$sELL'8"94S>VZrr)pathlibrtypingrrrrrrXultralytics.datar ultralytics.data.augmentr r r ultralytics.models.yolo.detectr ultralytics.utilsrr__all__rrrrrrsE00 (CC=+ VKVrm(mr