vis4d.eval.nuscenes.track3d

NuScenes 3D tracking evaluation code.

Classes

NuScenesTrack3DEvaluator([metadata])

NuScenes 3D tracking evaluation class.
class NuScenesTrack3DEvaluator(metadata=('use_camera',))[source]

NuScenes 3D tracking evaluation class.

Initialize NuScenes evaluator.

__repr__()[source]

Concise representation of the dataset evaluator.

Return type:

str

evaluate(metric)[source]

Evaluate the results.

Return type:

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

gather(gather_func)[source]

Gather variables in case of distributed setting (if needed).

Parameters:

gather_func (Callable[[Any], Any]) – Gather function.

Return type:

None

process_batch(tokens, boxes_3d, velocities, class_ids, scores_3d, track_ids)[source]

Process the results.

Return type:

None

reset()[source]

Reset evaluator.

Return type:

None

save(metric, output_dir)[source]

Save the results to json files.

Return type:

None

property metrics: list[str]

Supported metrics.