Bỏ để qua phần nội dung

Tài liệu tham khảo cho ultralytics/models/fastsam/val.py

Ghi

Tệp này có sẵn tại https://github.com/ultralytics/ultralytics/blob/main/ultralytics/Mô hình/fastsam/val.py. Nếu bạn phát hiện ra một vấn đề, vui lòng giúp khắc phục nó bằng cách đóng góp Yêu cầu 🛠️ kéo. Cảm ơn bạn 🙏 !



ultralytics.models.fastsam.val.FastSAMValidator

Căn cứ: SegmentationValidator

Lớp xác thực tùy chỉnh để nhanh chóng SAM (Mô hình phân đoạn bất cứ điều gì) phân đoạn trong Ultralytics YOLO khuôn khổ.

Mở rộng lớp SegmentationValidator, tùy chỉnh quy trình xác thực cụ thể để nhanh chóng SAM. Lớp học này đặt nhiệm vụ thành 'phân đoạn' và sử dụng SegmentMetrics để đánh giá. Ngoài ra, các tính năng vẽ bị vô hiệu hóa để tránh sai sót trong quá trình xác thực.

Thuộc tính:

Tên Kiểu Sự miêu tả
dataloader

Đối tượng data loader được sử dụng để xác thực.

save_dir str

Thư mục nơi kết quả xác thực sẽ được lưu.

pbar str

Một đối tượng thanh tiến trình.

args str

Đối số bổ sung để tùy chỉnh.

_callbacks str

Danh sách các hàm callback sẽ được gọi trong quá trình xác thực.

Mã nguồn trong 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)

Khởi tạo lớp FastSAMValidator, đặt tác vụ thành 'phân đoạn' và số liệu thành SegmentMetrics.

Thông số:

Tên Kiểu Sự miêu tả Mặc định
dataloader DataLoader

Dataloader được sử dụng để xác thực.

None
save_dir Path

Thư mục để lưu kết quả.

None
pbar tqdm

Thanh tiến trình để hiển thị tiến trình.

None
args SimpleNamespace

Cấu hình cho trình xác thực.

None
_callbacks dict

Từ điển để lưu trữ các hàm callback khác nhau.

None
Ghi chú

Biểu đồ cho ConfusionMatrix và các số liệu liên quan khác bị vô hiệu hóa trong lớp này để tránh lỗi.

Mã nguồn trong 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)