a yf@s:ddlZddlmZddlmZdgZGdddeZdS)N)DetectionValidator)ops NASValidatorc@seZdZdZddZdS)ra Ultralytics YOLO NAS Validator for object detection. Extends `DetectionValidator` from the Ultralytics models package and is designed to post-process the raw predictions generated by YOLO NAS models. It performs non-maximum suppression to remove overlapping and low-confidence boxes, ultimately producing the final detections. Attributes: args (Namespace): Namespace containing various configurations for post-processing, such as confidence and IoU. lb (torch.Tensor): Optional tensor for multilabel NMS. Example: ```python from ultralytics import NAS model = NAS("yolo_nas_s") validator = model.validator # Assumes that raw_preds are available final_preds = validator.postprocess(raw_preds) ``` Note: This class is generally not instantiated directly but is used internally within the `NAS` class. c Csht|dd}t||ddfdddd}tj||jj|jj|j d|jj pZ|jj |jj ddS)z4Apply Non-maximum suppression to prediction outputs.rFg?)labelsZ multi_labelZagnosticmax_detZ max_time_img) rZ xyxy2xywhtorchcatZpermuteZnon_max_suppressionargsconfZiouZlbZ single_clsZ agnostic_nmsr )selfZpreds_inZboxespredsrV/var/www/html/django/DPS/env/lib/python3.9/site-packages/ultralytics/models/nas/val.py postprocess%s"zNASValidator.postprocessN)__name__ __module__ __qualname____doc__rrrrrr s)r Zultralytics.models.yolo.detectrZultralytics.utilsr__all__rrrrrs