Reference for ultralytics/models/rtdetr/val.py
Note
Full source code for this file is available at https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/rtdetr/val.py. Help us fix any issues you see by submitting a Pull Request 🛠️. Thank you 🙏!
ultralytics.models.rtdetr.val.RTDETRDataset
Bases: YOLODataset
Source code in ultralytics/models/rtdetr/val.py
build_transforms(hyp=None)
Temporary, only for evaluation.
Source code in ultralytics/models/rtdetr/val.py
load_image(i, rect_mode=False)
ultralytics.models.rtdetr.val.RTDETRValidator
Bases: DetectionValidator
A class extending the DetectionValidator class for validation based on an RT-DETR detection model.
Example
Source code in ultralytics/models/rtdetr/val.py
46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 |
|
build_dataset(img_path, mode='val', batch=None)
Build an RTDETR Dataset.
Parameters:
Name | Type | Description | Default |
---|---|---|---|
img_path |
str
|
Path to the folder containing images. |
required |
mode |
str
|
|
'val'
|
batch |
int
|
Size of batches, this is for |
None
|
Source code in ultralytics/models/rtdetr/val.py
postprocess(preds)
Apply Non-maximum suppression to prediction outputs.
Source code in ultralytics/models/rtdetr/val.py
update_metrics(preds, batch)
Metrics.