vedo_normal_rendering.py

import vedo
import numpy as np

# Load or create a mesh
mesh = vedo.Sphere(res=50).triangulate()

# Compute normals
mesh.compute_normals()
normals = mesh.pointdata["Normals"]

# Map normals from [-1,1] to [0,1] for RGBA coloring
normals_col = (normals + 1) / 2.0
normals_col = ((normals + 1) / 2.0 * 255).astype(np.uint8)
normals_col = np.concatenate([normals_col, np.ones_like(normals_col[:, :1])*255], axis=-1)
mesh.pointcolors = normals_col

# Create a Plotter
plt = vedo.Plotter(offscreen=True)  # offscreen allows saving without GUI
plt.show(mesh, "Mesh with Normal Map Coloring", axes=1)

# # Save screenshot
plt.screenshot("mesh_normals.png")
plt.close()