fix example_00

Author: griegler

example/00_create_data/create_data.py

@@ -30,7 +30,7 @@
 dense[:,:,1] = 1
 dense[:,:,vx_res-2] = 1
 val_range = np.array([(0.9, 1.1)], dtype=np.float32)
-oc_from_dense1 = pyoctnet.Octree.create_from_dense(dense, val_range, n_threads=n_threads)
+oc_from_dense1 = pyoctnet.Octree.create_from_dense(dense[np.newaxis], n_threads=n_threads)
 oc_from_dense2 = pyoctnet.Octree.create_from_dense2(dense, dense[np.newaxis,...].copy(), n_threads=n_threads)
 
 vis.write_ply_voxels('dense.ply', dense)

example/00_create_data/vis.py

@@ -16,7 +16,7 @@ def write_ply_pcl(out_path, xyz, color=[128,128,128], color_array=None):
   f.write('property uchar blue\n')
   f.write('end_header\n')
 
-  for row in xrange(xyz.shape[0]):
+  for row in range(xyz.shape[0]):
     xyz_row = xyz[row]
     if color_array is not None:
       c = color_array[row]
@@ -27,7 +27,10 @@ def write_ply_pcl(out_path, xyz, color=[128,128,128], color_array=None):
 
 
 def write_ply_boxes(out_path, bxs, binary=False):
-  f = open(out_path, 'wb')
+  if binary:
+    f = open(out_path, 'wb')
+  else:
+    f = open(out_path, 'w')
   f.write("ply\n");
   if binary:
     f.write("format binary_little_endian 1.0\n")
@@ -118,9 +121,9 @@ def write_ply_boxes(out_path, bxs, binary=False):
 def write_ply_voxels(out_path, grid, color=[128,128,128], color_fcn=None, explode=1, binary=False):
   bxs = []
   width = 1
-  for d in xrange(grid.shape[0]):
-    for h in xrange(grid.shape[1]):
-      for w in xrange(grid.shape[2]):
+  for d in range(grid.shape[0]):
+    for h in range(grid.shape[1]):
+      for w in range(grid.shape[2]):
         if grid[d,h,w] == 0:
           continue
         x = w * explode