open3d.ml.torch.datasets.Waymo

class open3d.ml.torch.datasets.Waymo(dataset_path, name='Waymo', cache_dir='./logs/cache', use_cache=False, val_split=3, **kwargs)

Waymo 3D dataset for Object Detection, used in visualizer, training, or test

__init__(dataset_path, name='Waymo', cache_dir='./logs/cache', use_cache=False, val_split=3, **kwargs)

Initialize :param dataset_path: path to the dataset :type dataset_path: str :param kwargs:

static get_label_to_names()

Returns a label to names dictonary object.

Returns

A dict where keys are label numbers and values are the corresponding names.

get_split(split)

Returns a dataset split.

Parameters
  • split – A string identifying the dataset split that is usually one of

  • 'test', 'validation', or 'all'. ('training',) –

Returns

A dataset split object providing the requested subset of the data.

get_split_list(split)
is_tested()

Checks whether a datum has been tested.

Parameters

attr – The attributes associated with the datum.

Returns

This returns True if the test result has been stored for the datum with the specified attribute; else returns False.

static read_calib(path)
static read_label(path, calib)
static read_lidar(path)
save_test_result()

Saves the output of a model.

Parameters
  • results – The output of a model for the datum associated with the attribute passed.

  • attr – The attributes that correspond to the outputs passed in results.