Unverified Commit e41d2dd5 authored by wawatt's avatar wawatt Committed by GitHub
Browse files

Update my_detection_dataset

parent 83ff8a7e
......@@ -36,7 +36,7 @@ class myDetectionDataset(Dataset):
def __getitem__(self, idx):
scan_name = self.scan_names[idx]
point_cloud = np.load(os.path.join(self.data_path, scan_name)+'_seg.npz')['arr_0'] # Nx3
point_cloud = np.load(os.path.join(self.data_path, scan_name)+'_seg.npy') # Nx3
bbox = np.load(os.path.join(self.data_path, scan_name)+'_pos.npy') # K,20
if not self.use_color:
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment