bug in dataloader

Author: AllenXiangX

utils/data_loaders.py

@@ -210,8 +210,8 @@ def __init__(self, cfg):
     def get_dataset(self, subset):
         file_list = self._get_file_list(self.cfg, self._get_subset(subset))
         transforms = self._get_transforms(self.cfg, subset)
-        # required_items = ['partial_cloud'] if subset == DatasetSubset.TEST else ['partial_cloud', 'gtcloud']
-        required_items = ['partial_cloud', 'gtcloud']
+        required_items = ['partial_cloud'] if subset == DatasetSubset.TEST else ['partial_cloud', 'gtcloud']
+
         return Dataset({
             'required_items': required_items,
             'shuffle': subset == DatasetSubset.TRAIN
@@ -253,7 +253,7 @@ def _get_transforms(self, cfg, subset):
         else:
             return utils.data_transforms.Compose([{
                 'callback': 'ToTensor',
-                'objects': ['partial_cloud', 'gtcloud']
+                'objects': ['partial_cloud']
             }])
 
     def _get_subset(self, subset):