You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
124 lines
3.8 KiB
124 lines
3.8 KiB
import open3d as o3d |
|
from open3d.visualization.gui import Application |
|
from open3d.visualization import O3DVisualizer |
|
import sys, os |
|
|
|
|
|
# 假设 obj_file_path 是你的 OBJ 文件路径，image_save_path 是保存图片的路径 |
|
filePath = "G://obj_fix/print" |
|
pid = "8160" |
|
obj_file_path = f"{filePath}/{pid}/{pid}.obj" |
|
image_save_path = f"{filePath}/{pid}/{pid}_test.png" |
|
|
|
|
|
|
|
# import open3d as o3d |
|
|
|
# 加载OBJ文件 |
|
mesh = o3d.io.read_triangle_mesh(obj_file_path) |
|
|
|
# 创建一个OffscreenRenderer对象 |
|
renderer = o3d.visualization.rendering.OffscreenRenderer() |
|
|
|
# 创建一个Open3DScene对象 |
|
scene = o3d.visualization.rendering.Open3DScene() |
|
|
|
# 将几何体添加到场景中 |
|
scene.add_geometry(mesh) |
|
|
|
# 创建一个材质并应用到几何体上 |
|
material = o3d.visualization.rendering.MaterialRecord() |
|
material.base_color = [1.0, 1.0, 1.0] # 白色材质 |
|
mesh.materials.append(material) |
|
|
|
# 创建一个灯光对象 |
|
light = o3d.visualization.rendering.Light() |
|
light.type = "directional" # 方向光 |
|
light.direction = [0.0, -1.0, 0.0] # 光照方向 |
|
light.color = [1.0, 1.0, 1.0] # 白色光线 |
|
|
|
# 将灯光添加到场景中 |
|
scene.add_light(light) |
|
|
|
# 设置渲染参数 |
|
renderer.set_scene(scene) |
|
|
|
# 渲染场景 |
|
image = renderer.render() |
|
|
|
# 显示渲染结果 |
|
o3d.geometry.image.axial_slice(image, 2, (128, 128, 128)) |
|
|
|
# # 计算顶点法线，这对于渲染是非常重要的 |
|
# mesh.compute_vertex_normals() |
|
|
|
# # 可选：可以对网格进行进一步处理，例如滤波或下采样 |
|
# # mesh = o3d.geometry.filter_triangle_mesh(mesh, "default") |
|
|
|
# # 创建一个Open3D的可视化对象 |
|
# vis = o3d.visualization.Visualizer() |
|
# vis.create_window() |
|
|
|
# # 将网格添加到可视化对象中 |
|
# vis.add_geometry(mesh) |
|
|
|
# # 创建一个环境光源 |
|
# light = get_light_source("default") |
|
# light.set_intensity([1.0, 1.0, 1.0]) # 设置光照强度 |
|
|
|
# # 将灯光添加到可视化对象中 |
|
# vis.add_light(light) |
|
|
|
# # 调整灯光位置和方向 |
|
# light.set_position([0.0, 0.0, 0.0]) |
|
# light.set_direction([0.0, 0.0, 1.0]) |
|
# screenshot = vis.capture_screen() |
|
# o3d.io.write_image("screenshot.png", screenshot) |
|
|
|
# # 渲染并显示窗口 |
|
# vis.run() |
|
|
|
|
|
# # 加载一个点云或者网格 |
|
# mesh = o3d.io.read_triangle_mesh(obj_file_path) |
|
|
|
# # 创建 O3DVisualizer 实例 |
|
# vis = o3d.visualization.O3DVisualizer("O3D Visualizer Example", 1024, 768) |
|
|
|
# # 添加几何体到可视化窗口 |
|
# vis.add_geometry("Mesh", mesh) |
|
|
|
# # 显示可视化窗口 |
|
# Application.instance.add_window(vis) |
|
# Application.instance.run() |
|
|
|
# # Derive the object path set the model, material, and shader |
|
# model_dir = sys.argv[1] |
|
# model_name = os.path.join(model_dir, os.path.basename(model_dir) + ".obj") |
|
# model = o3d.io.read_triangle_mesh(obj_file_path) |
|
# material = o3d.visualization.rendering.Material() |
|
# material.shader = "defaultLit" |
|
|
|
# # Derive the texture paths |
|
# albedo_name = os.path.join(model_dir, "albedo.png") |
|
# normal_name = os.path.join(model_dir, "normal.png") |
|
# ao_name = os.path.join(model_dir, "ao.png") |
|
# metallic_name = os.path.join(model_dir, "metallic.png") |
|
# roughness_name = os.path.join(model_dir, "roughness.png") |
|
|
|
# # Check if the textures are available and loads the texture. For example, if metallic exists then load metallic texture |
|
# if os.path.exists(albedo_name): |
|
# material.albedo_img = o3d.io.read_image(albedo_name) |
|
# if os.path.exists(normal_name): |
|
# material.normal_img = o3d.io.read_image(normal_name) |
|
# if os.path.exists(ao_name): |
|
# material.ao_img = o3d.io.read_image(ao_name) |
|
# if os.path.exists(metallic_name): |
|
# material.base_metallic = 1.0 |
|
# material.metallic_img = o3d.io.read_image(metallic_name) |
|
# if os.path.exists(roughness_name): |
|
# material.roughness_img = o3d.io.read_image(roughness_name) |
|
|
|
# # Draw an object named cube using the available model and texture |
|
# o3d.visualization.draw([{"name": "cube", "geometry": model, "material": material}]) |
|
|
|
|