Extract UV map as .png file about trimesh HOT 1 OPEN

import numpy as np
import cv2
from scipy.interpolate import griddata
 
def compute_interpolation_map(shape, tcoords, values):
    points = (tcoords * np.asarray(shape)[None, :]).astype(np.int32)
    x = np.arange(shape[0])
    y = np.flip(np.arange(shape[1]))
    X, Y = np.meshgrid(x, y)
    res = griddata(points, values, (X, Y),  method='cubic')
    res[np.isnan(res)] = 0
    return res

tcoords = ... # you already have them
vertex_colors = ... # I assume you already have them
image_size = (512, 512) # create a 512 x 512 texture
texture_map = compute_interpolation_map(image_size, tcoords, vertex_colors) 
cv2.imwrite(image_path, texture_map)