如何使用PyTorch3D将顶点颜色（特征）转至UV空间纹理贴图？

已知前提数据 #

• 3D模型的顶点坐标集合 vertices、面顶点索引集合 faces
• UV纹理坐标集合 uv、UV面索引集合 uv_faces
• 与顶点一一对应的颜色特征集合 vertex colors

具体实现步骤 #

• 初始化纹理画布
  创建指定分辨率的空白纹理贴图（例如1024×1024），画布每个像素初始化为透明或默认底色。

• 遍历模型面，关联UV与顶点颜色
  逐个处理每个3D面：

  1. 通过faces拿到当前面的顶点索引，匹配对应顶点的vertex colors；
  2. 通过uv_faces拿到当前面对应的UV坐标索引，提取出对应的UV点；
  3. 将UV坐标转换为纹理画布的像素坐标：UV的(0,0)对应画布左下角，(1,1)对应右上角，计算每个UV点的像素位置（例如像素X = UV.x * 纹理宽度，像素Y = (1 - UV.y) * 纹理高度，适配图像坐标系）。

• 面内像素的颜色插值填充
  针对当前UV面构成的三角形（四边形需拆分为两个三角形）：
  使用重心坐标插值法，对三角形覆盖的每一个像素，计算其在UV三角形内的重心权重，以此对三个顶点的颜色进行加权求和，得到该像素的最终颜色并填充到画布。
  也可采用扫描线算法遍历三角形区域，批量计算并填充像素，提升效率。

• 处理UV重叠区域
  若同一纹理像素对应多个3D顶点，可根据需求选择处理规则：

  • 取颜色平均值
  • 保留权重最高的顶点颜色
  • 采用覆盖规则（如最后绘制的颜色覆盖之前的）

• 导出纹理贴图
  将填充完成的画布导出为PNG、TGA等标准纹理格式。

效果示例说明 #

示例效果为：纹理贴图的色彩与3D模型表面顶点颜色的渐变完全匹配，每个纹理区域的颜色都是对应模型面内顶点颜色的平滑插值结果，呈现出自然的色彩过渡。

内容的提问来源于stack exchange，提问作者XiaoJin