delete texture colors

app.py

@@ -46,7 +46,7 @@ description = '''
 </th>
 </table>
 
-#### The reconstruction + refinement + video take about 80~120 seconds for single image.
+#### The reconstruction + refinement + video take about 80~200 seconds for single image.
 
 <details>
 

apps/infer.py

@@ -50,10 +50,9 @@ def generate_model(in_path, model_type):
     
     torch.cuda.empty_cache()
 
-    config_dict = {'loop_smpl': 50,
-                   'loop_cloth': 100,
+    config_dict = {'loop_smpl': 100,
+                   'loop_cloth': 200,
                    'patience': 5,
-                   'vis_freq': 10,
                    'out_dir': './results',
                    'hps_type': 'pymaf',
                    'config': f"./configs/{model_type}.yaml"}
@@ -414,12 +413,17 @@ def generate_model(in_path, model_type):
                 mesh_pr.faces_packed().detach().squeeze(0).cpu(),
                 process=False, maintains_order=True
             )
-            final_colors = query_color(
-                mesh_pr.verts_packed().detach().squeeze(0).cpu(),
-                mesh_pr.faces_packed().detach().squeeze(0).cpu(),
-                in_tensor["image"],
-                device=device,
-            )
+            
+            # with front texture
+            # final_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(0).detach().cpu() + 1.0) * 0.5 * 255.0
             final.visual.vertex_colors = final_colors
             final.export(
                 f"{config_dict['out_dir']}/{cfg.name}/obj/{data['name']}_refine.obj")
@@ -454,6 +458,7 @@ def generate_model(in_path, model_type):
     for element in dir():
         if 'path' not in element:
             del locals()[element]
+            
     torch.cuda.empty_cache()
     
     return [smpl_path, smpl_path, smpl_npy_path, recon_path, recon_path, refine_path, refine_path, video_path, overlap_path]