处理染色问题

libs/MVS/display_demo.py

@@ -0,0 +1,77 @@
+import open3d as o3d
+import os
+import argparse
+import numpy as np
+from PIL import Image
+
+class DisplayProcessor:
+    def __init__(self):
+
+        # argv = sys.argv[sys.argv.index("--") + 1:] if "--" in sys.argv else []
+        parser = argparse.ArgumentParser()
+
+        parser.add_argument(
+            "--id", 
+            required=True, 
+        )
+        parser.add_argument(
+            "--mask",
+            default=0
+        )
+
+        args = parser.parse_args()
+
+        self.id = args.id
+        mask = args.mask
+        out_dir = f"out.{self.id}.nomask"
+        if mask==1:
+            out_dir = f"out.{self.id}"
+
+        self.mesh = None
+        self.obj_path = f"/home/algo/Documents/openMVS/data/{self.id}/{out_dir}/mesh.obj"
+        
+        self.load_and_show()
+        
+    def load_and_show(self):
+        # 加载并变换所有模型
+
+        # 加载网格
+        mesh = None
+        try:
+            mesh = o3d.io.read_triangle_mesh(self.obj_path, enable_post_processing=True)
+            if not mesh.has_vertices():
+                print(f"警告: 网格无有效顶点 - {self.obj_path}")
+        except Exception as e:
+            print(f"加载模型失败: {self.obj_path} - {e}")
+
+        if not mesh:
+            print("没有加载到任何模型，请检查错误信息")
+        else:
+            # 可视化模型
+            print("显示模型... (按'Q'退出)")
+
+            try:
+                from packaging import version
+                o3d_version = version.parse(o3d.__version__)
+                # 新版本 draw_geometries 参数
+                if o3d_version >= version.parse("0.13.0"):
+                    o3d.visualization.draw_geometries(
+                        [mesh],
+                        window_name="模型展示",
+                        mesh_show_back_face=True,
+                        mesh_show_wireframe=False
+                    )
+                # 旧版本 draw_geometries 参数
+                else:
+                    o3d.visualization.draw_geometries(
+                        [mesh],
+                        window_name="模型展示",
+                        point_show_normal=False,
+                        mesh_show_back_face=True
+                    )
+            except Exception as e:
+                print(f"使用 draw_geometries 可视化失败: {e}")
+
+# 主程序
+if __name__ == "__main__":
+    DisplayProcessor().load_and_show()