vis4d.eval.scalabel.detect

Scalabel detection evaluator.

Classes

ScalabelDetectEvaluator(annotation_path[, ...])

Scalabel 2D detection evaluation class.
class ScalabelDetectEvaluator(annotation_path, config=None, mask_threshold=0.0)[source]

Scalabel 2D detection evaluation class.

__init__(annotation_path, config=None, mask_threshold=0.0)[source]

Initialize the evaluator.

__repr__()[source]

Concise representation of the dataset evaluator.

Return type:

str

property metrics: list[str]

Supported metrics.

process_batch(frame_ids, sample_names, sequence_names, pred_boxes, pred_classes, pred_scores, pred_masks=None)[source]

Process tracking results.

Return type:

None

evaluate(metric)[source]

Evaluate the dataset.

Return type:

tuple[Dict[str, Union[float, int, Tensor]], str]