Refactor mesh export process in inference.py to directly save as GLB format, removing intermediate OBJ conversion steps. Update return value to reflect new output format.

inference.py

@@ -73,16 +73,27 @@ def generate3d(model, rgb, ccm, device):
 
     start_time = time.time()
     with torch.no_grad():
-        # First export to OBJ to get the mesh with textures
-        mesh_path_obj = tempfile.NamedTemporaryFile(suffix=f"", delete=False).name
-        model.export_mesh(data_config, mesh_path_obj, tri_fea_2 = triplane_feature2)
-        
-        # Load the OBJ mesh and convert to GLB
-        mesh = Mesh.load(mesh_path_obj+".obj", bound=0.9, front_dir="+z")
         mesh_path_glb = tempfile.NamedTemporaryFile(suffix=f"", delete=False).name
-        mesh.write(mesh_path_glb+".glb")
+        model.export_mesh(data_config, mesh_path_glb, tri_fea_2 = triplane_feature2)
+
+        # glctx = dr.RasterizeGLContext()#dr.RasterizeCudaContext()
+        # mesh_path_obj = tempfile.NamedTemporaryFile(suffix=f"", delete=False).name
+        # model.export_mesh_wt_uv(glctx, data_config, mesh_path_obj, "", device, res=(1024,1024), tri_fea_2=triplane_feature2)
+
+        # mesh = Mesh.load(mesh_path_obj+".obj", bound=0.9, front_dir="+z")
+        # mesh_path_glb = tempfile.NamedTemporaryFile(suffix=f"", delete=False).name
+        # mesh.write(mesh_path_glb+".glb")
+
+        # # mesh_obj2 = trimesh.load(mesh_path_glb+".glb", file_type='glb')
+        # # mesh_path_obj2 = tempfile.NamedTemporaryFile(suffix=f"", delete=False).name
+        # # mesh_obj2.export(mesh_path_obj2+".obj")
+
+        # with zipfile.ZipFile(mesh_path_obj+'.zip', 'w') as myzip:
+        #     myzip.write(mesh_path_obj+'.obj', mesh_path_obj.split("/")[-1]+'.obj')
+        #     myzip.write(mesh_path_obj+'.png', mesh_path_obj.split("/")[-1]+'.png')
+        #     myzip.write(mesh_path_obj+'.mtl', mesh_path_obj.split("/")[-1]+'.mtl')
 
     end_time = time.time()
     elapsed_time = end_time - start_time
     print(f"uv takes {elapsed_time}s")
-    return mesh_path_glb+".glb"
+    return mesh_path_glb+".obj"