Spaces:
Paused
Paused
File size: 2,183 Bytes
5b58e07 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 |
import os
import pymeshlab as ml
input_dir = "/public/home/wangshuo/gap/assembly/data/car_1k/obj"
output_base_dir = "/public/home/wangshuo/gap/assembly/data/car_1k/pose_4view"
import os
import pymeshlab as ml
import numpy as np
# 四个视角的旋转角度
rotations = [
{"x": -90, "y": 0},
{"x": -90, "y": 90},
{"x": -90, "y": 180},
{"x": -90, "y": 270},
]
# 旋转矩阵计算函数
def get_rotation_matrix(x_angle, y_angle):
# 计算绕 x 轴的旋转矩阵
x_rot_matrix = np.array([
[1, 0, 0],
[0, np.cos(np.radians(x_angle)), -np.sin(np.radians(x_angle))],
[0, np.sin(np.radians(x_angle)), np.cos(np.radians(x_angle))]
])
# 计算绕 y 轴的旋转矩阵
y_rot_matrix = np.array([
[np.cos(np.radians(y_angle)), 0, np.sin(np.radians(y_angle))],
[0, 1, 0],
[-np.sin(np.radians(y_angle)), 0, np.cos(np.radians(y_angle))]
])
# 合成旋转矩阵（先绕 x 轴旋转，再绕 y 轴旋转）
rotation_matrix = np.dot(y_rot_matrix, x_rot_matrix)
return rotation_matrix
# 遍历每个 obj 文件
for filename in os.listdir(input_dir):
if filename.endswith(".obj"):
obj_path = os.path.join(input_dir, filename)
base_name = os.path.splitext(filename)[0]
# 创建渲染输出目录
for i, rotation in enumerate(rotations):
output_dir = os.path.join(output_base_dir, f"view_{i+1}")
os.makedirs(output_dir, exist_ok=True)
# 加载网格
ms = ml.MeshSet()
ms.load_new_mesh(obj_path)
# 获取旋转矩阵
rot_matrix = get_rotation_matrix(rotation["x"], rotation["y"])
# 应用旋转矩阵
ms.apply_filter("transform_apply", matrix=rot_matrix.flatten().tolist())
# 渲染并保存图片
output_image_path = os.path.join(output_dir, f"{base_name}.png")
ms.save_current_mesh_image(output_image_path, image_width=800, image_height=600)
print(f"Rendered {filename} at rotation x={rotation['x']}°, y={rotation['y']}° -> {output_image_path}")
print("All OBJ files rendered.")
|