open3d.ml.torch.datasets

Classes

Argoverse(dataset_path[, info_path, name, …])

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

Custom3D(dataset_path[, name, cache_dir, …])

A template for customized datasetthat you can use with a dataloader to feed data when training a model.

InferenceDummySplit(inference_data)

KITTI(dataset_path[, name, cache_dir, …])

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

Lyft(dataset_path[, info_path, name, …])

Lyft level 5 dataset for Object Detection, used in visualizer, training, or test

NuScenes(dataset_path[, info_path, name, …])

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

ParisLille3D(dataset_path[, name, …])

This class is used to create a dataset based on the ParisLille3D dataset, and used in visualizer, training, or testing. The ParisLille3D dataset is best used to train models for urban infrastructure.

S3DIS(dataset_path[, name, cache_dir, …])

This class is used to create a dataset based on the S3DIS (Stanford Large-Scale 3D Indoor Spaces) dataset, and used in visualizer, training, or testing.

SemSegRandomSampler(dataset)

Random sampler for semantic segmentation datsets

SemSegSpatiallyRegularSampler(dataset)

Spatially regularSampler sampler for semantic segmentation datsets

Semantic3D(dataset_path[, name, cache_dir, …])

This class is used to create a dataset based on the Semantic3D dataset, and used in visualizer, training, or testing.

SemanticKITTI(dataset_path[, name, …])

This class is used to create a dataset based on the SemanticKitti dataset, and used in visualizer, training, or testing.

ShapeNet(dataset_path[, name, …])

Toronto3D(dataset_path[, name, cache_dir, …])

This class is used to create a dataset based on the Toronto3D dataset, and used in visualizer, training, or testing.

Waymo(dataset_path[, name, cache_dir, …])

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