cloudViewer.ml.torch.dataloaders.ConcatBatcher#

class cloudViewer.ml.torch.dataloaders.ConcatBatcher(device, model='KPConv')[source]#

ConcatBatcher for KPConv.

__init__(device, model='KPConv')[source]#

Initialize.

Parameters:: device – torch device ‘gpu’ or ‘cpu’
Returns:: The corresponding class.
Return type:: class

collate_fn(batches)[source]#

Collate function called by original PyTorch dataloader.

Parameters:: batches – a batch of data
Returns:: the batched result
Return type:: class