Dataset for whole pointclouds with farthest point sampling _incomplete_

2020-05-19 17:15:01 +02:00
parent fcd5ee4d29
commit 196b1af7ae
3 changed files with 31 additions and 1 deletions
--- a/_templates/new_project/datasets/template_dataset.py
+++ b/_templates/new_project/datasets/template_dataset.py
@ -3,4 +3,10 @@ from torch.utils.data import Dataset

 class TemplateDataset(Dataset):
    def __init__(self, *args, **kwargs):
-        super(TemplateDataset, self).__init__()
+        super(TemplateDataset, self).__init__()
+
+    def __len__(self):
+        pass
+
+    def __getitem__(self, item):
+        return item