rot

gradio_demo.py

@@ -89,7 +89,8 @@ def mesh_gen(tmp_dir, simplify, num_inference_steps):
 
     mesh = trimesh.load_mesh(f"{tmp_dir}/mesh.ply")
     vertex_normals = mesh.vertex_normals
-    theta = np.radians(90)  # Rotation angle in radians
+    theta = np.radians(180)  # Rotation angle in radians
+    # Create rotation matrix
     cos_theta = np.cos(theta)
     sin_theta = np.sin(theta)
     rotation_matrix = np.array([
@@ -98,9 +99,10 @@ def mesh_gen(tmp_dir, simplify, num_inference_steps):
         [0, 0, 1]
     ])
     rotated_normal = np.dot(vertex_normals, rotation_matrix.T)
-    rotated_normal = rotated_normal / np.linalg.norm(rotated_normal)
+    # rotated_normal = rotated_normal / np.linalg.norm(rotated_normal)
     colors = (-rotated_normal + 1) / 2.0
-    colors = (colors * 255).astype(np.uint8)  # Convert to 8-bit color
+    # colors = (-vertex_normals + 1) / 2.0
+    colors = (colors * 255).clip(0, 255).astype(np.uint8)  # Convert to 8-bit color
     # print(colors.shape)
     mesh.visual.vertex_colors = colors[..., [2, 1, 0]]  # RGB -> BGR
     mesh.export(f"{tmp_dir}/mesh_normal.ply", file_type="ply")