|h!ddlmZddlmZmZddlZddlmZddlm Z ddl m Z ddl m Z ddlmZd d lmZd d lmZGd d eZy))Path)AnyDictN)Model)DEFAULT_CFG_DICT)attempt_download_asset) torch_load) model_info) NASPredictor) NASValidatorc eZdZdZd deddffd Zd deddfdZdded edeee ffd Z e deeeee fffd Z xZ S)NASa YOLO-NAS model for object detection. This class provides an interface for the YOLO-NAS models and extends the `Model` class from Ultralytics engine. It is designed to facilitate the task of object detection using pre-trained or custom-trained YOLO-NAS models. Attributes: model (torch.nn.Module): The loaded YOLO-NAS model. task (str): The task type for the model, defaults to 'detect'. predictor (NASPredictor): The predictor instance for making predictions. validator (NASValidator): The validator instance for model validation. Methods: info: Log model information and return model details. Examples: >>> from ultralytics import NAS >>> model = NAS("yolo_nas_s") >>> results = model.predict("ultralytics/assets/bus.jpg") Notes: YOLO-NAS models only support pre-trained models. Do not provide YAML configuration files. modelreturnNcdt|jdvsJdt| |dy)z.yml.yamlz0YOLO-NAS models only support pre-trained models.detect)taskN)rsuffixsuper__init__)selfr __class__s [/var/www/html/test/engine/venv/lib/python3.12/site-packages/ultralytics/models/nas/model.pyrz NAS.__init__+s4E{!!)::n.new_forwardAs:://2 2rcjSN)r)verbosers rzNAS._load..Is tzzr cy)NFr/rrr,zNAS._load..Lsrr)T)super_gradientsrrr rrtrainingmodelsgetforwardr$fusetorchtensorstridedict enumerate _class_namesnamesis_fusedyamlpt_pathrr overridesr&eval)rrrr0rr(s` r_loadz NAS._load0s g%% U?#$:7$CDDJ r\(1188<r\s5 *.>04!QR%QRr