Zum Inhalt springen

Referenz fĂŒr ultralytics/models/fastsam/val.py

Hinweis

Diese Datei ist verfĂŒgbar unter https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/models/ fastsam/val .py. Wenn du ein Problem entdeckst, hilf bitte mit, es zu beheben, indem du einen Pull Request đŸ› ïž einreichst. Vielen Dank 🙏!



ultralytics.models.fastsam.val.FastSAMValidator

Basen: SegmentationValidator

Benutzerdefinierte Validierungsklasse fĂŒr die schnelle SAM (Segment Anything Model) Segmentierung im Ultralytics YOLO Framework.

Erweitert die Klasse SegmentationValidator und passt den Validierungsprozess speziell fĂŒr schnelle SAM an. Diese Klasse setzt die Aufgabe auf "Segment" und verwendet die SegmentMetrics fĂŒr die Auswertung. Außerdem werden die Plotting-Funktionen deaktiviert um Fehler bei der Validierung zu vermeiden.

Attribute:

Name Typ Beschreibung
dataloader

Das fĂŒr die Validierung verwendete Data Loader-Objekt.

save_dir str

Das Verzeichnis, in dem die Validierungsergebnisse gespeichert werden.

pbar str

Ein Fortschrittsbalken-Objekt.

args str

ZusĂ€tzliche Argumente fĂŒr Anpassungen.

_callbacks str

Liste der Callback-Funktionen, die wÀhrend der Validierung aufgerufen werden sollen.

Quellcode in ultralytics/models/fastsam/val.py
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)

Initialisiere die Klasse FastSAMValidator, indem du die Aufgabe auf "Segment" und die Metrik auf "SegmentMetrics" setzt.

Parameter:

Name Typ Beschreibung Standard
dataloader DataLoader

Dataloader, der fĂŒr die Validierung verwendet wird.

None
save_dir Path

Verzeichnis zum Speichern der Ergebnisse.

None
pbar tqdm

Fortschrittsbalken zur Anzeige des Fortschritts.

None
args SimpleNamespace

Konfiguration fĂŒr den Validator.

None
_callbacks dict

Wörterbuch, um verschiedene Callback-Funktionen zu speichern.

None
Anmerkungen

Plots fĂŒr ConfusionMatrix und andere verwandte Metriken sind in dieser Klasse deaktiviert, um Fehler zu vermeiden.

Quellcode in ultralytics/models/fastsam/val.py
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)





Erstellt 2023-11-12, aktualisiert 2024-06-02
Autoren: glenn-jocher (5), Burhan-Q (1)