rotated_normal

gradio_demo.py

@@ -99,7 +99,7 @@ def mesh_gen(tmp_dir, simplify, num_inference_steps):
     ])
     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 = (-rotated_normal + 1) / 2.0
     colors = (colors * 255).astype(np.uint8)  # Convert to 8-bit color
     # print(colors.shape)
     mesh.visual.vertex_colors = colors[..., [2, 1, 0]]  # RGB -> BGR