सामग्री पर जाएं

के लिए संदर्भ ultralytics/models/nas/model.py

नोट

यह फ़ाइल यहाँ उपलब्ध है https://github.com/ultralytics/ultralytics/बूँद/मुख्य/ultralytics/मॉडल/NAS/model.py. यदि आप कोई समस्या देखते हैं तो कृपया पुल अनुरोध का योगदान करके इसे ठीक करने में मदद करें 🛠️। 🙏 धन्यवाद !



ultralytics.models.nas.model.NAS

का रूप: Model

YOLO ऑब्जेक्ट डिटेक्शन के लिए NAS मॉडल।

यह वर्ग के लिए एक इंटरफ़ेस प्रदान करता है YOLO-NAS मॉडल और विस्तार Model से वर्ग Ultralytics इंजन। यह पूर्व-प्रशिक्षित या कस्टम-प्रशिक्षित का उपयोग करके वस्तु का पता लगाने के कार्य को सुविधाजनक बनाने के लिए डिज़ाइन किया गया है YOLO-NAS मॉडल।

उदाहरण
from ultralytics import NAS

model = NAS('yolo_nas_s')
results = model.predict('ultralytics/assets/bus.jpg')

विशेषताएँ:

नाम प्रकार या क़िस्‍म
model str

पूर्व-प्रशिक्षित मॉडल या मॉडल नाम का पथ। ' के लिए चूकyolo_nas_s.पीटी'।

नोट

YOLO-NAS मॉडल केवल पूर्व-प्रशिक्षित मॉडल का समर्थन करते हैं। YAML कॉन्फ़िगरेशन फ़ाइलें प्रदान न करें।

में स्रोत कोड ultralytics/models/nas/model.py
class NAS(Model):
    """
    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.

    Example:
        ```python
        from ultralytics import NAS

        model = NAS('yolo_nas_s')
        results = model.predict('ultralytics/assets/bus.jpg')
        ```

    Attributes:
        model (str): Path to the pre-trained model or model name. Defaults to 'yolo_nas_s.pt'.

    Note:
        YOLO-NAS models only support pre-trained models. Do not provide YAML configuration files.
    """

    def __init__(self, model="yolo_nas_s.pt") -> None:
        """Initializes the NAS model with the provided or default 'yolo_nas_s.pt' model."""
        assert Path(model).suffix not in {".yaml", ".yml"}, "YOLO-NAS models only support pre-trained models."
        super().__init__(model, task="detect")

    @smart_inference_mode()
    def _load(self, weights: str, task: str):
        """Loads an existing NAS model weights or creates a new NAS model with pretrained weights if not provided."""
        import super_gradients

        suffix = Path(weights).suffix
        if suffix == ".pt":
            self.model = torch.load(weights)
        elif suffix == "":
            self.model = super_gradients.training.models.get(weights, pretrained_weights="coco")
        # Standardize model
        self.model.fuse = lambda verbose=True: self.model
        self.model.stride = torch.tensor([32])
        self.model.names = dict(enumerate(self.model._class_names))
        self.model.is_fused = lambda: False  # for info()
        self.model.yaml = {}  # for info()
        self.model.pt_path = weights  # for export()
        self.model.task = "detect"  # for export()

    def info(self, detailed=False, verbose=True):
        """
        Logs model info.

        Args:
            detailed (bool): Show detailed information about model.
            verbose (bool): Controls verbosity.
        """
        return model_info(self.model, detailed=detailed, verbose=verbose, imgsz=640)

    @property
    def task_map(self):
        """Returns a dictionary mapping tasks to respective predictor and validator classes."""
        return {"detect": {"predictor": NASPredictor, "validator": NASValidator}}

task_map property

संबंधित भविष्यवक्ता और सत्यापनकर्ता वर्गों के लिए एक शब्दकोश मानचित्रण कार्यों देता है।

__init__(model='yolo_nas_s.pt')

प्रदान किए गए या डिफ़ॉल्ट के साथ NAS मॉडल को प्रारंभ करता है 'yolo_nas_s.pt' मॉडल।

में स्रोत कोड ultralytics/models/nas/model.py
def __init__(self, model="yolo_nas_s.pt") -> None:
    """Initializes the NAS model with the provided or default 'yolo_nas_s.pt' model."""
    assert Path(model).suffix not in {".yaml", ".yml"}, "YOLO-NAS models only support pre-trained models."
    super().__init__(model, task="detect")

info(detailed=False, verbose=True)

लॉग मॉडल जानकारी।

पैरामीटर:

नाम प्रकार या क़िस्‍म चूक
detailed bool

मॉडल के बारे में विस्तृत जानकारी दिखाएं।

False
verbose bool

वर्बोसिटी को नियंत्रित करता है।

True
में स्रोत कोड ultralytics/models/nas/model.py
def info(self, detailed=False, verbose=True):
    """
    Logs model info.

    Args:
        detailed (bool): Show detailed information about model.
        verbose (bool): Controls verbosity.
    """
    return model_info(self.model, detailed=detailed, verbose=verbose, imgsz=640)





Created 2023-11-12, Updated 2024-06-02
Authors: glenn-jocher (5), Burhan-Q (1)