Prueba generar Textura

app.py

@@ -21,7 +21,7 @@ from tqdm import tqdm
 
 from src.utils.camera_util import (FOV_to_intrinsics, get_circular_camera_poses, get_zero123plus_input_cameras)
 from src.utils.infer_util import (remove_background, resize_foreground)
-from src.utils.mesh_util import save_glb, save_obj
+from src.utils.mesh_util import save_glb, save_obj, save_texmap
 from src.utils.train_util import instantiate_from_config
 
 # Inicializa un tensor en CUDA y verifica el dispositivo.
@@ -87,7 +87,7 @@ def generate_mvs(input_image, sample_steps, sample_seed):
 
     return z123_image, show_image
 
-# Convierte imágenes en modelos 3D.
+# Convierte imágenes en modelos 3D y genera texturas.
 @spaces.GPU
 def make3d(images):
     
@@ -113,20 +113,22 @@ def make3d(images):
     mesh_basename = os.path.basename(mesh_fpath).split('.')[0]
     mesh_dirname = os.path.dirname(mesh_fpath)
     mesh_glb_fpath = os.path.join(mesh_dirname, f"{mesh_basename}.glb")
+    texmap_fpath = os.path.join(mesh_dirname, f"{mesh_basename}_tex.png")
 
     with torch.no_grad():
         planes = model.forward_planes(images, input_cameras)
-        mesh_out = model.extract_mesh(planes, use_texture_map=False, **infer_config)
+        mesh_out = model.extract_mesh(planes, use_texture_map=True, **infer_config)
 
-        vertices, faces, vertex_colors = mesh_out
+        vertices, faces, vertex_colors, texture_map = 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, texture_map, mesh_glb_fpath)
+        save_obj(vertices, faces, vertex_colors, texture_map, mesh_fpath)
+        save_texmap(texture_map, texmap_fpath)
 
-        print(f"Mesh saved to {mesh_fpath}")
+        print(f"Mesh and texture saved to {mesh_fpath} and {texmap_fpath}")
 
-    return mesh_fpath, mesh_glb_fpath
+    return mesh_fpath, mesh_glb_fpath, texmap_fpath
 
 # Procesa la imagen generada a partir de un prompt de texto.
 @spaces.GPU
@@ -240,6 +242,12 @@ with gr.Blocks() as demo:
                         label="(Formato GLB)",
                         interactive=False,
                     )
+                with gr.Tab("Textura"):
+                    output_texture = gr.Image(
+                        label="Textura Generada",
+                        type="pil",
+                        interactive=False
+                    )
 
     mv_images = gr.State()
 
@@ -254,7 +262,7 @@ with gr.Blocks() as demo:
     ).success(
         fn=make3d,
         inputs=[mv_images],
-        outputs=[output_model_obj, output_model_glb]
+        outputs=[output_model_obj, output_model_glb, output_texture]
     )
 
 demo.launch()