vis4d.eval.nuscenes.detect3d

NuScenes 3D detection evaluation code.

Classes

NuScenesDet3DEvaluator(data_root, version, split)

NuScenes 3D detection evaluation class.

class NuScenesDet3DEvaluator(data_root, version, split, save_only=False, class_map=None, metadata=('use_camera',), use_default_attr=False, velocity_thres=1.0)

NuScenes 3D detection evaluation class.

Initialize NuScenes evaluator.

__repr__()

Concise representation of the dataset evaluator.

Return type:

str

evaluate(metric)

Evaluate the results.

Return type:

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

gather(gather_func)

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

Parameters:

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

Return type:

None

get_attributes(name, velocity)

Get nuScenes attributes.

Return type:

str

process_batch(tokens, boxes_3d, velocities, class_ids, scores_3d, attributes=None)

Process the results.

Return type:

None

reset()

Reset evaluator.

Return type:

None

save(metric, output_dir)

Save the results to json files.

Return type:

None

property metrics: list[str]

Supported metrics.