Saltar al contenido

Referencia para ultralytics/models/fastsam/val.py

Nota

Este archivo est谩 disponible en https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/models/ fastsam/val .py. Si detectas alg煤n problema, por favor, ayuda a solucionarlo contribuyendo con una Pull Request 馃洜锔. 隆Gracias 馃檹!



ultralytics.models.fastsam.val.FastSAMValidator

Bases: SegmentationValidator

Clase de validaci贸n personalizada para la segmentaci贸n r谩pida SAM (Segment Anything Model) en Ultralytics YOLO framework.

Extiende la clase SegmentationValidator, personalizando el proceso de validaci贸n espec铆ficamente para SAM r谩pido. Esta clase establece la tarea en "segmentar" y utiliza las SegmentMetrics para la evaluaci贸n. Adem谩s, se desactivan las funciones de trazado para evitar errores durante la validaci贸n.

Atributos:

Nombre Tipo Descripci贸n
dataloader

El objeto cargador de datos utilizado para la validaci贸n.

save_dir str

El directorio donde se guardar谩n los resultados de la validaci贸n.

pbar str

Un objeto de barra de progreso.

args str

Argumentos adicionales para la personalizaci贸n.

_callbacks str

Lista de funciones de llamada de retorno que se invocar谩n durante la validaci贸n.

C贸digo fuente en 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)

Inicializa la clase FastSAMValidator, estableciendo la tarea como 'segmento' y las m茅tricas como SegmentMetrics.

Par谩metros:

Nombre Tipo Descripci贸n Por defecto
dataloader DataLoader

Cargador de datos que se utilizar谩 para la validaci贸n.

None
save_dir Path

Directorio para guardar los resultados.

None
pbar tqdm

Barra de progreso para mostrar el progreso.

None
args SimpleNamespace

Configuraci贸n del validador.

None
_callbacks dict

Diccionario para almacenar varias funciones de llamada de retorno.

None
Notas

Los gr谩ficos de la Matriz de Confusi贸n y otras m茅tricas relacionadas est谩n desactivados en esta clase para evitar errores.

C贸digo fuente en 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)





Creado 2023-11-12, Actualizado 2024-05-18
Autores: glenn-jocher (4), Burhan-Q (1)