add readme.md

utils/data_load.py

@@ -14,25 +14,16 @@ class DataLoadUtil:
     
     @staticmethod
     def load_exr_image(file_path):
-        # 打开 EXR 文件
         exr_file = OpenEXR.InputFile(file_path)
-        
-        # 获取 EXR 文件的头部信息，包括尺寸
         header = exr_file.header()
         dw = header['dataWindow']
         width = dw.max.x - dw.min.x + 1
         height = dw.max.y - dw.min.y + 1
-
-        # 定义通道，通常法线图像是 RGB
         float_channels = ['R', 'G', 'B']
-
-        # 读取 EXR 文件中的每个通道并转化为浮点数数组
         img_data = []
         for channel in float_channels:
             channel_data = exr_file.channel(channel, Imath.PixelType(Imath.PixelType.FLOAT))
             img_data.append(np.frombuffer(channel_data, dtype=np.float32).reshape((height, width)))
-
-        # 将各通道组合成一个 (height, width, 3) 的 RGB 图像
         img = np.stack(img_data, axis=-1)
         return img
 

utils/vis.py

@@ -128,10 +128,10 @@ class visualizeUtil:
 if __name__ == "__main__":
     root = r"C:\Document\Local Project\nbv_rec\nbv_reconstruction\temp"
     model_dir = r"H:\\AI\\Datasets\\scaled_object_box_meshes"
-    scene = "test_obj"
+    scene = "box"
     output_dir = r"C:\Document\Local Project\nbv_rec\nbv_reconstruction\test"
     
-    visualizeUtil.save_all_cam_pos_and_cam_axis(root, scene, output_dir)
+    #visualizeUtil.save_all_cam_pos_and_cam_axis(root, scene, output_dir)
     visualizeUtil.save_all_combined_pts(root, scene, output_dir)
     visualizeUtil.save_target_mesh_at_world_space(root, model_dir, scene)
     #visualizeUtil.save_points_and_normals(root, scene,"10", output_dir, binocular=True)