Vai al contenuto

Riferimento per ultralytics/models/fastsam/val.py

Nota

Questo file è disponibile su https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/models/ fastsam/val .py. Se riscontri un problema, contribuisci a risolverlo inviando una Pull Request 🛠️. Grazie 🙏!



ultralytics.models.fastsam.val.FastSAMValidator

Basi: SegmentationValidator

Classe di validazione personalizzata per la segmentazione veloce di SAM (Segment Anything Model) nel framework Ultralytics YOLO .

Estende la classe SegmentationValidator, personalizzando il processo di validazione in modo specifico per il veloce SAM. Questa classe imposta il compito su "segmento" e utilizza le SegmentMetrics per la valutazione. Inoltre, le funzioni di plottaggio sono disabilitate per evitare errori durante la validazione.

Attributi:

Nome Tipo Descrizione
dataloader

L'oggetto caricatore di dati utilizzato per la convalida.

save_dir str

La directory in cui verranno salvati i risultati della convalida.

pbar str

Un oggetto barra di avanzamento.

args str

Argomenti aggiuntivi per la personalizzazione.

_callbacks str

Elenco delle funzioni di callback da invocare durante la convalida.

Codice sorgente 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)

Inizializza la classe FastSAMValidator, impostando il task su 'segment' e le metriche su SegmentMetrics.

Parametri:

Nome Tipo Descrizione Predefinito
dataloader DataLoader

Dataloader da utilizzare per la convalida.

None
save_dir Path

Directory per salvare i risultati.

None
pbar tqdm

Barra di avanzamento per visualizzare i progressi.

None
args SimpleNamespace

Configurazione del validatore.

None
_callbacks dict

Dizionario per memorizzare le varie funzioni di callback.

None
Note

I grafici della matrice di confusione e di altre metriche correlate sono disabilitati in questa classe per evitare errori.

Codice sorgente 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)





Creato 2023-11-12, Aggiornato 2023-11-25
Autori: glenn-jocher (3)