From 2f597bdc85377bccb64c3e040c08fa207964a4f5 Mon Sep 17 00:00:00 2001 From: Ziyue Feng Date: Sun, 8 Mar 2020 17:11:19 -0400 Subject: [PATCH] Update stanford.py (#25) actually, the batch index is the first column --- demo/stanford.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/demo/stanford.py b/demo/stanford.py index 46fe871..3f89f6d 100644 --- a/demo/stanford.py +++ b/demo/stanford.py @@ -159,7 +159,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:] # last column is the batch index pred_pcd.points = o3d.utility.Vector3dVector(coordinates * config.voxel_size) pred_pcd.colors = o3d.utility.Vector3dVector(colors / 255)