Update scannet.py (#26)

actually, the batch index is the first column
chrischoy · Mar 8, 2020 · 5eba0bc · 5eba0bc
1 parent 1614faa
commit 5eba0bc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/demo/scannet.py b/demo/scannet.py
@@ -147,7 +147,7 @@ def generate_input_sparse_tensor(file_name, voxel_size=0.05):
 
   # Create a point cloud file
   pred_pcd = o3d.geometry.PointCloud()
-  coordinates = soutput.C.numpy()[:, :3]  # last column is the batch index
+  coordinates = soutput.C.numpy()[:, 1:]  # first column is the batch index
   pred_pcd.points = o3d.utility.Vector3dVector(coordinates * config.voxel_size)
   pred_pcd.colors = o3d.utility.Vector3dVector(colors / 255)