Перейти к содержимому

Ссылка для ultralytics/models/fastsam/val.py

Примечание

Этот файл доступен по адресу https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/models/ fastsam/val .py. Если ты заметил проблему, пожалуйста, помоги исправить ее, отправив Pull Request 🛠️. Спасибо 🙏!



ultralytics.models.fastsam.val.FastSAMValidator

Базы: SegmentationValidator

Пользовательский класс валидации для быстрой сегментации SAM (Segment Anything Model) во фреймворке Ultralytics YOLO .

Расширяет класс SegmentationValidator, настраивая процесс валидации специально для быстрых SAM. Этот класс устанавливает задачу на 'segment' и использует SegmentMetrics для оценки. Кроме того, отключаются функции построения графиков чтобы избежать ошибок во время валидации.

Атрибуты:

Имя Тип Описание
dataloader

Объект загрузчика данных, используемый для валидации.

save_dir str

Директория, в которую будут сохраняться результаты проверки.

pbar str

Объект прогресс-бара.

args str

Дополнительные аргументы в пользу кастомизации.

_callbacks str

Список функций обратного вызова, которые будут вызываться во время валидации.

Исходный код в 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)

Инициализируй класс FastSAMValidator, установив задачу на 'segment' и метрику на SegmentMetrics.

Параметры:

Имя Тип Описание По умолчанию
dataloader DataLoader

Dataloader, который будет использоваться для валидации.

None
save_dir Path

Директория для сохранения результатов.

None
pbar tqdm

Прогресс-бар для отображения прогресса.

None
args SimpleNamespace

Конфигурация для валидатора.

None
_callbacks dict

Словарь для хранения различных функций обратного вызова.

None
Примечания

Графики для ConfusionMatrix и других связанных с ней метрик в этом классе отключены, чтобы избежать ошибок.

Исходный код в 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)





Создано 2023-11-12, Обновлено 2023-11-25
Авторы: glenn-jocher (3)