Skip to content

Référence pour ultralytics/models/fastsam/val.py

Note

Ce fichier est disponible à l'adresse https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/models/ fastsam/val .py. Si tu repères un problème, aide à le corriger en contribuant à une Pull Request 🛠️. Merci 🙏 !



ultralytics.models.fastsam.val.FastSAMValidator

Bases : SegmentationValidator

Classe de validation personnalisée pour une segmentation rapide SAM (Segment Anything Model) dans le cadre de Ultralytics YOLO .

Étend la classe SegmentationValidator, en personnalisant le processus de validation spécifiquement pour SAM. Cette classe définit la tâche comme étant "segment" et utilise les SegmentMetrics pour l'évaluation. En outre, les fonctions de traçage sont désactivées pour éviter les erreurs lors de la validation.

Attributs :

Nom Type Description
dataloader

L'objet du chargeur de données utilisé pour la validation.

save_dir str

Le répertoire dans lequel les résultats de la validation seront enregistrés.

pbar str

Un objet de barre de progression.

args str

Arguments supplémentaires pour la personnalisation.

_callbacks str

Liste des fonctions de rappel Ă  invoquer pendant la validation.

Code source dans 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)

Initialise la classe FastSAMValidator, en fixant la tâche à "segment" et les métriques à SegmentMetrics.

Paramètres :

Nom Type Description DĂ©faut
dataloader DataLoader

Dataloader Ă  utiliser pour la validation.

None
save_dir Path

Répertoire pour enregistrer les résultats.

None
pbar tqdm

Barre de progression pour afficher la progression.

None
args SimpleNamespace

Configuration du validateur.

None
_callbacks dict

Dictionnaire permettant de stocker diverses fonctions de rappel.

None
Notes

Les tracés de la matrice de confusion et d'autres mesures connexes sont désactivés dans cette classe pour éviter les erreurs.

Code source dans 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)





Créé le 2023-11-12, Mis à jour le 2023-11-25
Auteurs : glenn-jocher (3)