انتقل إلى المحتوى

مرجع ل 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. هذه الفئة يضبط المهمة على "تقسيم" ويستخدم SegmentMetrics للتقييم. بالإضافة إلى ذلك ، يتم تعطيل ميزات التخطيط لتجنب الأخطاء أثناء التحقق من الصحة.

سمات:

اسم نوع وصف
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 313233 34353637383940
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)