add normal as color for better obbj visualization

apps/infer.py

@@ -295,6 +295,7 @@ def generate_model(in_path, model_type):
             process=False,
             maintains_order=True
         )
+        smpl_obj.visual.vertex_colors = (smpl_obj.vertex_normals+1.0)*255.0*0.5
         smpl_obj.export(
             f"{config_dict['out_dir']}/{cfg.name}/obj/{data['name']}_smpl.obj")
 
@@ -334,6 +335,7 @@ def generate_model(in_path, model_type):
         recon_obj = trimesh.Trimesh(
             verts_pr, faces_pr, process=False, maintains_order=True
         )
+        recon_obj.visual.vertex_colors = (recon_obj.vertex_normals+1.0)*255.0*0.5
         recon_obj.export(
             os.path.join(config_dict['out_dir'], cfg.name,
                          f"obj/{data['name']}_recon.obj")