Source code for catalyst.dl.callbacks.metrics.iou

from typing import List  # isort:skip

from catalyst.dl.core import MetricCallback, MultiMetricCallback
from catalyst.utils import criterion


def _get_default_classwise_iou_args(num_classes: int) -> List[int]:
    assert num_classes > 0, "num_classes must be greater than 0"
    return [str(i) for i in range(num_classes)]


[docs]class IouCallback(MetricCallback): """ IoU (Jaccard) metric callback. """
[docs] def __init__( self, input_key: str = "targets", output_key: str = "logits", prefix: str = "iou", eps: float = 1e-7, threshold: float = None, activation: str = "Sigmoid", ): """ Args: input_key (str): input key to use for iou calculation specifies our ``y_true``. output_key (str): output key to use for iou calculation; specifies our ``y_pred`` prefix (str): key to store in logs eps (float): epsilon to avoid zero division threshold (float): threshold for outputs binarization activation (str): An torch.nn activation applied to the outputs. Must be one of ['none', 'Sigmoid', 'Softmax2d'] """ super().__init__( prefix=prefix, metric_fn=criterion.iou, input_key=input_key, output_key=output_key, eps=eps, threshold=threshold, activation=activation )
JaccardCallback = IouCallback
[docs]class ClasswiseIouCallback(MultiMetricCallback): """ Classwise IoU (Jaccard) metric callback. """
[docs] def __init__( self, input_key: str = "targets", output_key: str = "logits", prefix: str = "iou", classes: List[str] = None, num_classes: int = None, eps: float = 1e-7, threshold: float = None, activation: str = "Sigmoid", ): """ Args: input_key (str): input key to use for iou calculation specifies our ``y_true``. output_key (str): output key to use for iou calculation; specifies our ``y_pred`` prefix (str): key to store in logs (will be prefix_class_name) classes (List[str]): list of class names You should specify either 'classes' or 'num_classes' num_classes (int): number of classes You should specify either 'classes' or 'num_classes' eps (float): epsilon to avoid zero division threshold (float): threshold for outputs binarization activation (str): An torch.nn activation applied to the outputs. Must be one of ['none', 'Sigmoid', 'Softmax2d'] """ assert classes is not None or num_classes is not None, \ "You should specify either 'classes' or 'num_classes'" list_args = classes or _get_default_classwise_iou_args(num_classes) super().__init__( prefix=prefix, metric_fn=criterion.iou, list_args=list_args, input_key=input_key, output_key=output_key, eps=eps, threshold=threshold, activation=activation, )
ClasswiseJaccardCallback = ClasswiseIouCallback __all__ = [ "IouCallback", "JaccardCallback", "ClasswiseIouCallback", "ClasswiseJaccardCallback" ]