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

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

नोट

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



ultralytics.models.rtdetr.model.RTDETR

का रूप: Model

Baidu के लिए इंटरफ़ेस RT-DETR को गढ़ना। यह विजन ट्रांसफार्मर-आधारित ऑब्जेक्ट डिटेक्टर वास्तविक समय प्रदर्शन प्रदान करता है उच्च सटीकता के साथ। यह कुशल हाइब्रिड एन्कोडिंग, IoU-जागरूक क्वेरी चयन और अनुकूलनीय अनुमान गति का समर्थन करता है।

विशेषताएँ:

नाम प्रकार विवरण: __________
model str

पूर्व-प्रशिक्षित मॉडल के लिए पथ। 'rtdetr-l.pt' के लिए डिफ़ॉल्ट।

में स्रोत कोड ultralytics/models/rtdetr/model.py
18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 424344454647 4849505152535455 56
class RTDETR(Model):
    """
    Interface for Baidu's RT-DETR model. This Vision Transformer-based object detector provides real-time performance
    with high accuracy. It supports efficient hybrid encoding, IoU-aware query selection, and adaptable inference speed.

    Attributes:
        model (str): Path to the pre-trained model. Defaults to 'rtdetr-l.pt'.
    """

    def __init__(self, model="rtdetr-l.pt") -> None:
        """
        Initializes the RT-DETR model with the given pre-trained model file. Supports .pt and .yaml formats.

        Args:
            model (str): Path to the pre-trained model. Defaults to 'rtdetr-l.pt'.

        Raises:
            NotImplementedError: If the model file extension is not 'pt', 'yaml', or 'yml'.
        """
        if model and model.split(".")[-1] not in ("pt", "yaml", "yml"):
            raise NotImplementedError("RT-DETR only supports creating from *.pt, *.yaml, or *.yml files.")
        super().__init__(model=model, task="detect")

    @property
    def task_map(self) -> dict:
        """
        Returns a task map for RT-DETR, associating tasks with corresponding Ultralytics classes.

        Returns:
            dict: A dictionary mapping task names to Ultralytics task classes for the RT-DETR model.
        """
        return {
            "detect": {
                "predictor": RTDETRPredictor,
                "validator": RTDETRValidator,
                "trainer": RTDETRTrainer,
                "model": RTDETRDetectionModel,
            }
        }

task_map: dict property

के लिए एक कार्य मानचित्र लौटाता है RT-DETR, संबंधित के साथ कार्यों को जोड़ना Ultralytics वर्गों।

देता:

नाम प्रकार विवरण: __________
dict dict

एक शब्दकोश मैपिंग कार्य नाम Ultralytics के लिए कार्य कक्षाएं RT-DETR को गढ़ना।

__init__(model='rtdetr-l.pt')

इनक्वियलाइज़ करता है RT-DETR दिए गए पूर्व-प्रशिक्षित मॉडल फ़ाइल के साथ मॉडल। .pt और .yaml स्वरूपों का समर्थन करता है।

पैरामीटर:

नाम प्रकार विवरण: __________ चूक
model str

पूर्व-प्रशिक्षित मॉडल के लिए पथ। 'rtdetr-l.pt' के लिए डिफ़ॉल्ट।

'rtdetr-l.pt'

उठाती:

प्रकार विवरण: __________
NotImplementedError

यदि मॉडल फ़ाइल एक्सटेंशन 'pt', 'yaml', या 'yml' नहीं है।

में स्रोत कोड ultralytics/models/rtdetr/model.py
def __init__(self, model="rtdetr-l.pt") -> None:
    """
    Initializes the RT-DETR model with the given pre-trained model file. Supports .pt and .yaml formats.

    Args:
        model (str): Path to the pre-trained model. Defaults to 'rtdetr-l.pt'.

    Raises:
        NotImplementedError: If the model file extension is not 'pt', 'yaml', or 'yml'.
    """
    if model and model.split(".")[-1] not in ("pt", "yaml", "yml"):
        raise NotImplementedError("RT-DETR only supports creating from *.pt, *.yaml, or *.yml files.")
    super().__init__(model=model, task="detect")





2023-11-12 बनाया गया, अपडेट किया गया 2023-11-25
लेखक: ग्लेन-जोचर (3)