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).