SIGMitch commited on
Commit
c935f69
·
verified ·
1 Parent(s): 806fbdd
Files changed (1) hide show
  1. app.py +5 -14
app.py CHANGED
@@ -228,24 +228,15 @@ def make3d(images):
228
  # get mesh
229
  mesh_out = model.extract_mesh(
230
  planes,
231
- use_texture_map=True,
232
  **infer_config,
233
  )
234
- vertices, faces, uvs, mesh_tex_idx, tex_map = mesh_out
235
- save_obj_with_mtl(
236
- vertices.data.cpu().numpy(),
237
- uvs.data.cpu().numpy(),
238
- faces.data.cpu().numpy(),
239
- mesh_tex_idx.data.cpu().numpy(),
240
- tex_map.permute(1, 2, 0).data.cpu().numpy(),
241
- mesh_fpath,
242
- )
243
 
244
- #vertices, faces, vertex_colors = mesh_out
245
- #vertices = vertices[:, [1, 2, 0]]
246
 
247
- #save_glb(vertices, faces, vertex_colors, mesh_glb_fpath)
248
- #save_obj(vertices, faces, vertex_colors, mesh_fpath)
249
 
250
  print(f"Mesh saved to {mesh_fpath}")
251
 
 
228
  # get mesh
229
  mesh_out = model.extract_mesh(
230
  planes,
231
+ use_texture_map=False,
232
  **infer_config,
233
  )
 
 
 
 
 
 
 
 
 
234
 
235
+ vertices, faces, vertex_colors = mesh_out
236
+ vertices = vertices[:, [1, 2, 0]]
237
 
238
+ save_glb(vertices, faces, vertex_colors, mesh_glb_fpath)
239
+ save_obj(vertices, faces, vertex_colors, mesh_fpath)
240
 
241
  print(f"Mesh saved to {mesh_fpath}")
242