vis4d.eval.shift.seg

SHIFT segmentation evaluator.

Classes

SHIFTSegEvaluator([ignore_classes_as_cityscapes])

SHIFT segmentation evaluation class.
class SHIFTSegEvaluator(ignore_classes_as_cityscapes=True)[source]

SHIFT segmentation evaluation class.

Initialize the evaluator.

__repr__()[source]

Concise representation of the dataset evaluator.

Return type:

str

process_batch(prediction, groundtruth)[source]

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