Overslaan naar inhoud

Referentie voor ultralytics/models/fastsam/val.py

Opmerking

Dit bestand is beschikbaar op https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/models/ fastsam/val .py. Als je een probleem ziet, help het dan oplossen door een Pull Request 🛠️ bij te dragen. Bedankt 🙏!



ultralytics.models.fastsam.val.FastSAMValidator

Basis: SegmentationValidator

Aangepaste validatieklasse voor snelle SAM (Segment Anything Model) segmentatie in Ultralytics YOLO framework.

Breidt de klasse SegmentationValidator uit en past het validatieproces specifiek aan voor snelle SAM. Deze klasse stelt de taak in op 'segment' en gebruikt de SegmentMetrics voor evaluatie. Bovendien worden plotfuncties uitgeschakeld om fouten tijdens de validatie te voorkomen.

Kenmerken:

Naam Type Beschrijving
dataloader

Het gegevensladerobject dat wordt gebruikt voor validatie.

save_dir str

De map waar de validatieresultaten worden opgeslagen.

pbar str

Een voortgangsbalkobject.

args str

Extra argumenten voor aanpassing.

_callbacks str

Lijst met callbackfuncties die worden aangeroepen tijdens de validatie.

Broncode 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)

Initialiseer de klasse FastSAMValidator en stel de taak in op 'segment' en de metriek op SegmentMetrics.

Parameters:

Naam Type Beschrijving Standaard
dataloader DataLoader

Dataloader die moet worden gebruikt voor validatie.

None
save_dir Path

Directory om resultaten op te slaan.

None
pbar tqdm

Voortgangsbalk voor het weergeven van de voortgang.

None
args SimpleNamespace

Configuratie voor de validator.

None
_callbacks dict

Woordenboek om verschillende callbackfuncties in op te slaan.

None
Opmerkingen

Plots voor ConfusionMatrix en andere gerelateerde statistieken zijn uitgeschakeld in deze klasse om fouten te voorkomen.

Broncode 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)





Aangemaakt 2023-11-12, Bijgewerkt 2024-06-02
Auteurs: glenn-jocher (5), Burhan-Q (1)