Update open3d interface in completion example visualization (#184)

* Update resnet.py remove dimension from `MinkowskiGlobalMaxPooling` * update open3d api
NVIDIA · Jul 23, 2020 · 6b6a2df · 6b6a2df
1 parent 423f526
commit 6b6a2df
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/completion.py b/examples/completion.py
@@ -612,11 +612,11 @@ def visualize(net, dataloader, device, config):
             pcd = PointCloud(coords)
             pcd.estimate_normals()
             pcd.translate([0.6 * config.resolution, 0, 0])
-            pcd.rotate(M)
+            pcd.rotate(M, np.array([[0.0],[0.0],[0.0]]))
             opcd = PointCloud(data_dict['cropped_coords'][b])
             opcd.translate([-0.6 * config.resolution, 0, 0])
             opcd.estimate_normals()
-            opcd.rotate(M)
+            opcd.rotate(M, np.array([[0.0],[0.0],[0.0]]))
             o3d.visualization.draw_geometries([pcd, opcd])
 
             n_vis += 1