Upload ./svrm/ldm/vis_util.py with huggingface_hub

svrm/ldm/vis_util.py

@@ -26,6 +26,13 @@ from pytorch3d.renderer import (
     BlendParams,
 )
 
+import trimesh
+def load_glb_as_meshes(glb_filename, device):
+    mesh = trimesh.load(glb_filename, force='mesh')
+    vertices = torch.tensor(mesh.vertices, dtype=torch.float32).to(device)
+    faces = torch.tensor(mesh.faces, dtype=torch.int64).to(device)
+    return Meshes(verts=[vertices], faces=[faces])
+
 
 def render_func(
     obj_filename, 
@@ -46,7 +53,9 @@ def render_func(
         rgb: if set true, will convert result to rgb image/frame
     '''
     # load mesh
-    mesh = load_objs_as_meshes([obj_filename], device=device)
+    if obj_filename.endswith(".obj"): mesh = load_objs_as_meshes([obj_filename], device=device)
+    elif obj_filename.endswith(".glb"): mesh = load_glb_as_meshes(obj_filename, device)
+    else: raise f"only support obj or glb"
     meshes = mesh.extend(n_views)
     
     if azim is None: