コンテンツへスキップ

参考 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

Ultralytics YOLO フレームワークで、SAM (Segment Anything Model) セグメンテーションを高速に行うためのカスタム検証クラス。

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

バリデーションに使用するデータローダ。

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)