Update RigNet/quick_start.py

RigNet/quick_start.py

@@ -63,7 +63,6 @@ def create_single_data(mesh_filename):
     mesh_v = np.asarray(mesh.vertices)
     mesh_vn = np.asarray(mesh.vertex_normals)
     mesh_f = np.asarray(mesh.triangles)
-
     mesh_v, translation_normalize, scale_normalize = normalize_obj(mesh_v)
     mesh_normalized = o3d.geometry.TriangleMesh(vertices=o3d.utility.Vector3dVector(mesh_v), triangles=o3d.utility.Vector3iVector(mesh_f))
     o3d.io.write_triangle_mesh(mesh_filename.replace("_remesh.obj", "_normalized.obj"), mesh_normalized)
@@ -73,17 +72,17 @@ def create_single_data(mesh_filename):
     v = torch.from_numpy(v).float()
 
     # topology edges
-    print("     gathering topological edges.")
+    print("  gathering topological edges.")
     tpl_e = get_tpl_edges(mesh_v, mesh_f).T
     tpl_e = torch.from_numpy(tpl_e).long()
     tpl_e, _ = add_self_loops(tpl_e, num_nodes=v.size(0))
 
     # surface geodesic distance matrix
-    print("     calculating surface geodesic matrix.")
+    print("  calculating surface geodesic matrix.")
     surface_geodesic = calc_surface_geodesic(mesh)
 
     # geodesic edges
-    print("     gathering geodesic edges.")
+    print("  gathering geodesic edges.")
     geo_e = get_geo_edges(surface_geodesic, mesh_v).T
     geo_e = torch.from_numpy(geo_e).long()
     geo_e, _ = add_self_loops(geo_e, num_nodes=v.size(0))
@@ -91,16 +90,33 @@ def create_single_data(mesh_filename):
     # batch
     batch = torch.zeros(len(v), dtype=torch.long)
 
-    # voxel
-    if not os.path.exists(mesh_filename.replace('_remesh.obj', '_normalized.binvox')):
+    # voxel - FIXED: Use absolute path and better error handling
+    binvox_file = mesh_filename.replace('_remesh.obj', '_normalized.binvox')
+    normalized_obj = mesh_filename.replace("_remesh.obj", "_normalized.obj")
+    
+    if not os.path.exists(binvox_file):
+        print(f"  voxelizing mesh with binvox...")
+        
+        # Use absolute path to binvox (installed in Dockerfile)
         if platform == "linux" or platform == "linux2":
-            os.system("./binvox -d 88 -pb " + mesh_filename.replace("_remesh.obj", "_normalized.obj"))
+            cmd = f"binvox -d 88 -pb {normalized_obj}"
         elif platform == "win32":
-            os.system("binvox.exe -d 88 " + mesh_filename.replace("_remesh.obj", "_normalized.obj"))
+            cmd = f"binvox.exe -d 88 {normalized_obj}"
         else:
             raise Exception('Sorry, we currently only support windows and linux.')
-
-    with open(mesh_filename.replace('_remesh.obj', '_normalized.binvox'), 'rb') as fvox:
+        
+        print(f"  Running: {cmd}")
+        exit_code = os.system(cmd)
+        
+        if exit_code != 0:
+            raise Exception(f"binvox command failed with exit code {exit_code}. Command: {cmd}")
+        
+        if not os.path.exists(binvox_file):
+            raise Exception(f"binvox did not create output file: {binvox_file}")
+        
+        print(f"  ✓ Voxelization complete: {binvox_file}")
+    
+    with open(binvox_file, 'rb') as fvox:
         vox = binvox_rw.read_as_3d_array(fvox)
 
     data = Data(x=v[:, 3:6], pos=v[:, 0:3], tpl_edge_index=tpl_e, geo_edge_index=geo_e, batch=batch)