visualize with normal texture, download with half-colored texture

apps/infer.py

@@ -418,13 +418,24 @@ def generate_model(in_path, model_type):
                 mesh_pr.faces_packed().detach().squeeze(0).cpu(),
                 process=False, maintains_order=True
             )
+            
+            # only with front texture
+            tex_colors = query_color(
+                mesh_pr.verts_packed().detach().squeeze(0).cpu(),
+                mesh_pr.faces_packed().detach().squeeze(0).cpu(),
+                in_tensor["image"],
+                device=device,
+            )
 
-            # without front texture
-            final_colors = (mesh_pr.verts_normals_padded().squeeze(
+            # full normal textures
+            norm_colors = (mesh_pr.verts_normals_padded().squeeze(
                 0).detach().cpu() + 1.0) * 0.5 * 255.0
-            final.visual.vertex_colors = final_colors
+                
+            final.visual.vertex_colors = tex_colors
             final.export(
                 f"{config_dict['out_dir']}/{cfg.name}/obj/{data['name']}_refine.obj")
+                
+            final.visual.vertex_colors = norm_colors
             final.export(
                 f"{config_dict['out_dir']}/{cfg.name}/obj/{data['name']}_refine.glb")