asas

app.py

@@ -19,7 +19,7 @@ from src.utils.camera_util import (
     get_zero123plus_input_cameras,
     get_circular_camera_poses,
 )
-from src.utils.mesh_util import save_obj, save_glb
+from src.utils.mesh_util import save_obj, save_glb, save_obj_with_mtl
 from src.utils.infer_util import remove_background, resize_foreground, images_to_video
 
 import tempfile
@@ -228,15 +228,24 @@ def make3d(images):
         # get mesh
         mesh_out = model.extract_mesh(
             planes,
-            use_texture_map=False,
+            use_texture_map=True,
             **infer_config,
         )
+        vertices, faces, uvs, mesh_tex_idx, tex_map = mesh_out
+                save_obj_with_mtl(
+                vertices.data.cpu().numpy(),
+                uvs.data.cpu().numpy(),
+                faces.data.cpu().numpy(),
+                mesh_tex_idx.data.cpu().numpy(),
+                tex_map.permute(1, 2, 0).data.cpu().numpy(),
+                mesh_fpath,
+        )
 
-        vertices, faces, vertex_colors = mesh_out
-        vertices = vertices[:, [1, 2, 0]]
+        #vertices, faces, vertex_colors = mesh_out
+        #vertices = vertices[:, [1, 2, 0]]
         
-        save_glb(vertices, faces, vertex_colors, mesh_glb_fpath)
-        save_obj(vertices, faces, vertex_colors, mesh_fpath)
+        #save_glb(vertices, faces, vertex_colors, mesh_glb_fpath)
+        #save_obj(vertices, faces, vertex_colors, mesh_fpath)
         
         print(f"Mesh saved to {mesh_fpath}")