Update RigNet/quick_start.py

RigNet/quick_start.py

@@ -90,31 +90,29 @@ def create_single_data(mesh_filename):
     # batch
     batch = torch.zeros(len(v), dtype=torch.long)
 
-    # 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)
+        print(" voxelizing mesh with binvox...")
+        # On headless Linux (Hugging Face), run binvox under Xvfb
         if platform == "linux" or platform == "linux2":
-            cmd = f"binvox -d 88 -pb {normalized_obj}"
+            # -a will pick a free display (e.g. :99) and start Xvfb for this command
+            cmd = f"xvfb-run -a binvox -d 88 -pb {normalized_obj}"
         elif platform == "win32":
             cmd = f"binvox.exe -d 88 {normalized_obj}"
         else:
-            raise Exception('Sorry, we currently only support windows and linux.')
-        
-        print(f"  Running: {cmd}")
+            raise Exception("Sorry, we currently only support Windows and Linux.")
+    
+        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}")
+    
+        print(f" ✓ Voxelization complete: {binvox_file}")
     
     with open(binvox_file, 'rb') as fvox:
         vox = binvox_rw.read_as_3d_array(fvox)