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

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

नोट

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



ultralytics.models.fastsam.val.FastSAMValidator

का रूप: SegmentationValidator

तेजी से के लिए कस्टम सत्यापन वर्ग SAM (सेगमेंट कुछ भी मॉडल) विभाजन में Ultralytics YOLO ढांचा।

SegmentationValidator वर्ग का विस्तार करता है, विशेष रूप से तेजी के लिए सत्यापन प्रक्रिया को अनुकूलित करता है SAM. यह वर्ग कार्य को 'सेगमेंट' पर सेट करता है और मूल्यांकन के लिए सेगमेंटमेट्रिक्स का उपयोग करता है। साथ ही, प्लॉटिंग सुविधाएँ अक्षम हैं सत्यापन के दौरान त्रुटियों से बचने के लिए।

विशेषताएँ:

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

सत्यापन के लिए उपयोग किया जाने वाला डेटा लोडर ऑब्जेक्ट।

save_dir str

निर्देशिका जहां सत्यापन परिणाम सहेजे जाएंगे।

pbar str

एक प्रगति पट्टी ऑब्जेक्ट।

args str

अनुकूलन के लिए अतिरिक्त तर्क।

_callbacks str

सत्यापन के दौरान लागू किए जाने वाले कॉलबैक कार्यों की सूची।

में स्रोत कोड ultralytics/models/fastsam/val.py
 7 बांग्लादेश 8  9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31323334353637383940
class FastSAMValidator(SegmentationValidator):
    """
    Custom validation class for fast SAM (Segment Anything Model) segmentation in Ultralytics YOLO framework.

    Extends the SegmentationValidator class, customizing the validation process specifically for fast SAM. This class
    sets the task to 'segment' and uses the SegmentMetrics for evaluation. Additionally, plotting features are disabled
    to avoid errors during validation.

    Attributes:
        dataloader: The data loader object used for validation.
        save_dir (str): The directory where validation results will be saved.
        pbar: A progress bar object.
        args: Additional arguments for customization.
        _callbacks: List of callback functions to be invoked during validation.
    """

    def __init__(self, dataloader=None, save_dir=None, pbar=None, args=None, _callbacks=None):
        """
        Initialize the FastSAMValidator class, setting the task to 'segment' and metrics to SegmentMetrics.

        Args:
            dataloader (torch.utils.data.DataLoader): Dataloader to be used for validation.
            save_dir (Path, optional): Directory to save results.
            pbar (tqdm.tqdm): Progress bar for displaying progress.
            args (SimpleNamespace): Configuration for the validator.
            _callbacks (dict): Dictionary to store various callback functions.

        Notes:
            Plots for ConfusionMatrix and other related metrics are disabled in this class to avoid errors.
        """
        super().__init__(dataloader, save_dir, pbar, args, _callbacks)
        self.args.task = "segment"
        self.args.plots = False  # disable ConfusionMatrix and other plots to avoid errors
        self.metrics = SegmentMetrics(save_dir=self.save_dir, on_plot=self.on_plot)

__init__(dataloader=None, save_dir=None, pbar=None, args=None, _callbacks=None)

FastSAMValidator क्लास को इनिशियलाइज़ करें, टास्क को 'सेगमेंट' और मेट्रिक्स को SegmentMetrics पर सेट करें।

पैरामीटर:

नाम प्रकार विवरण: __________ चूक
dataloader DataLoader

सत्यापन के लिए उपयोग किए जाने वाले डेटालोडर।

None
save_dir Path

परिणामों को बचाने के लिए निर्देशिका।

None
pbar tqdm

प्रगति प्रदर्शित करने के लिए प्रगति पट्टी।

None
args SimpleNamespace

सत्यापनकर्ता के लिए कॉन्फ़िगरेशन।

None
_callbacks dict

विभिन्न कॉलबैक कार्यों को संग्रहीत करने के लिए शब्दकोश।

None
नोट्स

त्रुटियों से बचने के लिए इस वर्ग में ConfusionMatrix और अन्य संबंधित मीट्रिक के लिए प्लॉट अक्षम हैं।

में स्रोत कोड ultralytics/models/fastsam/val.py
23 बांग्लादेश 24 25 26 27 28 29 30 31 32 33 34353637383940
def __init__(self, dataloader=None, save_dir=None, pbar=None, args=None, _callbacks=None):
    """
    Initialize the FastSAMValidator class, setting the task to 'segment' and metrics to SegmentMetrics.

    Args:
        dataloader (torch.utils.data.DataLoader): Dataloader to be used for validation.
        save_dir (Path, optional): Directory to save results.
        pbar (tqdm.tqdm): Progress bar for displaying progress.
        args (SimpleNamespace): Configuration for the validator.
        _callbacks (dict): Dictionary to store various callback functions.

    Notes:
        Plots for ConfusionMatrix and other related metrics are disabled in this class to avoid errors.
    """
    super().__init__(dataloader, save_dir, pbar, args, _callbacks)
    self.args.task = "segment"
    self.args.plots = False  # disable ConfusionMatrix and other plots to avoid errors
    self.metrics = SegmentMetrics(save_dir=self.save_dir, on_plot=self.on_plot)





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