Reference for ultralytics/models/nas/val.py#
This page is sourced from https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/nas/val.py. Have an improvement or example to add? Open a Pull Request — thank you! 🙏
Class ultralytics.models.nas.val.NASValidator#
NASValidator()Bases: DetectionValidator
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
| Name | Type | Description |
|---|---|---|
args | SimpleNamespace | Namespace containing various configurations for post-processing, such as confidence and IoU thresholds. |
Examples
>>> from ultralytics import NAS
>>> model = NAS("yolo_nas_s.pt")
>>> metrics = model.val(data="coco8.yaml") # runs NASValidator internallyThis class is generally not instantiated directly but is used internally within the NAS class.
ultralytics/models/nas/val.py
class NASValidator(DetectionValidator):
"""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 (SimpleNamespace): Namespace containing various configurations for post-processing, such as confidence and
IoU thresholds.
Examples:
>>> from ultralytics import NAS
>>> model = NAS("yolo_nas_s.pt")
>>> metrics = model.val(data="coco8.yaml") # runs NASValidator internally
Notes:
This class is generally not instantiated directly but is used internally within the NAS class.
"""