open3d.ml.torch.vis.DatasetModel#

class open3d.ml.torch.vis.DatasetModel(dataset, split, indices)#

The class used to manage a dataset model.

Parameters:
  • dataset – The 3D ML dataset to use. You can use the base dataset, sample datasets , or a custom dataset.

  • split – A string identifying the dataset split that is usually one of ‘training’, ‘test’, ‘validation’, or ‘all’.

  • indices – The indices to be used for the datamodel. This may vary based on the split used.

__init__(dataset, split, indices)#
is_loaded(name)#

Check if the data is loaded.

load(name, fail_if_no_space=False)#

Check if data is not loaded, and then load the data.

unload(name)#

Unload the data (if it was loaded earlier).