Salta para o conteúdo

Referência para ultralytics/models/fastsam/val.py

Nota

Este ficheiro está disponível em https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/models/ fastsam/val .py. Se detectares um problema, por favor ajuda a corrigi-lo contribuindo com um Pull Request 🛠️. Obrigado 🙏!



ultralytics.models.fastsam.val.FastSAMValidator

Bases: SegmentationValidator

Classe de validação personalizada para uma segmentação rápida de SAM (Segment Anything Model) na estrutura Ultralytics YOLO .

Estende a classe SegmentationValidator, personalizando o processo de validação especificamente para SAM rápido. Esta classe define a tarefa como 'segment' e usa o SegmentMetrics para avaliação. Além disso, os recursos de plotagem são desativados para evitar erros durante a validação.

Atributos:

Nome Tipo Descrição
dataloader

O objeto do carregador de dados utilizado para a validação.

save_dir str

O diretório onde os resultados da validação serão guardados.

pbar str

Um objeto de barra de progresso.

args str

Argumentos adicionais para a personalização.

_callbacks str

Lista de funções de retorno de chamada a invocar durante a validação.

Código fonte em 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 a classe FastSAMValidator, definindo a tarefa como 'segment' e as métricas como SegmentMetrics.

Parâmetros:

Nome Tipo Descrição Predefinição
dataloader DataLoader

Carregador de dados a utilizar para a validação.

None
save_dir Path

Diretório para guardar os resultados.

None
pbar tqdm

Barra de progresso para mostrar o progresso.

None
args SimpleNamespace

Configura o validador.

None
_callbacks dict

Dicionário para armazenar várias funções de retorno de chamada.

None
Notas

Os gráficos para ConfusionMatrix e outras métricas relacionadas são desactivados nesta classe para evitar erros.

Código fonte em 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)





Created 2023-11-12, Updated 2024-06-02
Authors: glenn-jocher (5), Burhan-Q (1)