rot

gradio_demo.py

@@ -89,10 +89,20 @@ 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
+    cos_theta = np.cos(theta)
+    sin_theta = np.sin(theta)
+    rotation_matrix = np.array([
+        [cos_theta, -sin_theta, 0],
+        [sin_theta, cos_theta, 0],
+        [0, 0, 1]
+    ])
+    rotated_normal = np.dot(vertex_normals, rotation_matrix.T)
+    rotated_normal = rotated_normal / np.linalg.norm(rotated_normal)
     colors = (-vertex_normals + 1) / 2.0
     colors = (colors * 255).astype(np.uint8)  # Convert to 8-bit color
-    print(colors.shape)
-    mesh.visual.vertex_colors = colors[..., [0, 1, 2]]
+    # print(colors.shape)
+    mesh.visual.vertex_colors = colors[..., [2, 1, 0]]  # RGB -> BGR
     mesh.export(f"{tmp_dir}/mesh_normal.ply", file_type="ply")
 
     color_path = ply_to_glb(f"{tmp_dir}/mesh.ply")