vis4d.eval.shift.seg

SHIFT segmentation evaluator.

Classes

SHIFTSegEvaluator([ignore_classes_as_cityscapes])

SHIFT segmentation evaluation class.

class SHIFTSegEvaluator(ignore_classes_as_cityscapes=True)

SHIFT segmentation evaluation class.

__init__(ignore_classes_as_cityscapes=True)

Initialize the evaluator.

__repr__()

Concise representation of the dataset evaluator.

Return type:

str

process_batch(prediction, groundtruth)

Process sample and update confusion matrix.

Parameters:

• prediction (Union[ndarray[Any, dtype[bool_]], ndarray[Any, dtype[float32]], ndarray[Any, dtype[float64]], ndarray[Any, dtype[int32]], ndarray[Any, dtype[int64]], ndarray[Any, dtype[uint8]], ndarray[Any, dtype[uint16]], ndarray[Any, dtype[uint32]]]) – Predictions of shape [N,C,…] or [N,…] with C* being any number if channels. Note, C is passed, the prediction is converted to target labels by applying the max operations along the second axis

• groundtruth (ndarray[Any, dtype[int64]]) – Groundtruth of shape [N_batch, …] type int

Return type:

None