open3d.ml.torch.dataloaders.TorchDataloader

class open3d.ml.torch.dataloaders.TorchDataloader(dataset=None, preprocess=None, transform=None, use_cache=True, steps_per_epoch=None, **kwargs)

Data loader for torch framework.

__init__(dataset=None, preprocess=None, transform=None, use_cache=True, steps_per_epoch=None, **kwargs)

Initialize

Parameters

  • dataset – ml3d dataset class.

  • dataset – model’s preprocess method.

  • devce – model’s transform mthod.

  • use_cache – whether to use cached preprocessed data.

  • steps_per_epch – steps per epoch. The step number will be the number of samples in the data if steps_per_epoch=None

  • kwargs –

Returns

The corresponding class.

Return type

class