support glb and more input

Anymate/utils/ui_utils.py

@@ -19,8 +19,8 @@ def visualize_results(mesh_file=None, joints=None, conns=None, skins=None):
         
     if mesh_file is not None:
         # Load the original mesh (in blue) with transparency
-        # original_mesh = trimesh.load(mesh_file)
-        original_mesh = obj2mesh(mesh_file)
+        original_mesh = trimesh.load(mesh_file)
+        # original_mesh = obj2mesh(mesh_file)
         if skins is not None:
             # pdb.set_trace()
             # Get per-vertex colors based on skinning weights
@@ -50,8 +50,8 @@ def visualize_results(mesh_file=None, joints=None, conns=None, skins=None):
 
             # face_colors = np.zeros((len(original_mesh.faces), 4))
 
-            # processed_mesh = trimesh.load(mesh_file)
-            processed_mesh = obj2mesh(mesh_file)
+            processed_mesh = trimesh.load(mesh_file)
+            # processed_mesh = obj2mesh(mesh_file)
             # Assign vertex colors from original_mesh to processed_mesh
             # Since they might have different number of vertices, we need to find closest vertices
             
@@ -218,7 +218,7 @@ def prepare_blender_file(normalized_mesh_file):
 
     folder = normalized_mesh_file.replace('object.obj', '')
     abs_folder = os.path.abspath(folder)
-    os.system(f"python Render.py --path {abs_folder}")
+    os.system(f"python Render.py --path '{abs_folder}'")
 
     blender_file = os.path.join(folder, 'blender_output.blend')
     while not os.path.exists(blender_file):
@@ -248,7 +248,8 @@ def process_input(mesh_file):
     # make folder for tmp files
     os.makedirs(f"Anymate/tmp/{mesh_file.split('/')[-1].replace('.obj', '')}", exist_ok=True)
 
-    normalized_mesh = normalize_mesh(obj2mesh(mesh_file))
+    normalized_mesh = normalize_mesh(trimesh.load(mesh_file))
+    # normalized_mesh = normalize_mesh(obj2mesh(mesh_file))
     normalized_mesh_file = f"Anymate/tmp/{mesh_file.split('/')[-1].replace('.obj', '')}/object.obj"
     normalized_mesh.export(normalized_mesh_file)
 
@@ -274,8 +275,8 @@ def get_result_connectivity(mesh_file, model, pc, joints):
     return get_connectivity(pc, joints, model, device=anymate_args.device, save=mesh_file.replace('object.obj', 'conns.pt'))
 
 def get_result_skinning(mesh_file, model, pc, joints, conns):
-    # mesh = trimesh.load(mesh_file)
-    mesh = obj2mesh(mesh_file)
+    mesh = trimesh.load(mesh_file)
+    # mesh = obj2mesh(mesh_file)
     # vertices = torch.from_numpy(mesh.vertices, device='cpu')
     # vertex_normals = torch.from_numpy(mesh.vertex_normals, device='cpu')
     # vertices = torch.cat([vertices, vertex_normals], dim=-1)