point_to_primitive/datasets/prim_clustered.py

8 lines
210 B
Python

from torch.utils.data import Dataset
from ._point_dataset import _Point_Dataset
class TemplateDataset(_Point_Dataset):
def __init__(self, *args, **kwargs):
super(TemplateDataset, self).__init__()