19 lines
381 B
Python
19 lines
381 B
Python
import pickle
|
|
from collections import defaultdict
|
|
|
|
import numpy as np
|
|
|
|
from ._point_dataset import _Point_Dataset
|
|
|
|
|
|
class FullCloudsDataset(_Point_Dataset):
|
|
|
|
split: str
|
|
|
|
def __init__(self, *args, setting='pc', **kwargs):
|
|
self.setting = setting
|
|
super(FullCloudsDataset, self).__init__(*args, **kwargs)
|
|
|
|
def __len__(self):
|
|
return len(self._files)
|