Merge branch 'master' of http://www.hofee.top:3000/hofee/nbv_reconstruction

configs/local/inference_config.yaml

@@ -6,16 +6,16 @@ runner:
     cuda_visible_devices: "0,1,2,3,4,5,6,7"
     
   experiment:
-    name: local_eval
+    name: local_full_eval
     root_dir: "experiments"
-    epoch: 555 # -1 stands for last epoch
+    epoch: 20 # -1 stands for last epoch
 
   test:
     dataset_list:
       - OmniObject3d_train
 
   blender_script_path: "/media/hofee/data/project/python/nbv_reconstruction/blender/data_renderer.py"
-  output_dir: "/media/hofee/data/project/python/nbv_reconstruction/sample_for_training/inference_result2"
+  output_dir: "/media/hofee/data/project/python/nbv_reconstruction/sample_for_training/inference_result_full"
   pipeline: nbv_reconstruction_pipeline
   
 dataset:

utils/render.py

@@ -3,6 +3,7 @@ import os
 import json
 import subprocess
 import tempfile
+import shutil
 from utils.data_load import DataLoadUtil
 from utils.reconstruction import ReconstructionUtil
 from utils.pts import PtsUtil
@@ -19,6 +20,8 @@ class RenderUtil:
                 "cam_pose": nO_to_world_pose.tolist(),
                 "scene_path": scene_path
             }
+            scene_info_path = os.path.join(scene_path, "scene_info.json")
+            shutil.copy(scene_info_path, os.path.join(temp_dir, "scene_info.json"))
             params_data_path = os.path.join(temp_dir, "params.json")
             with open(params_data_path, 'w') as f:
                 json.dump(params, f)
@@ -30,6 +33,9 @@ class RenderUtil:
                 print(result.stderr)
                 return None
             path = os.path.join(temp_dir, "tmp")
+            # ------  Debug Start ------
+            import ipdb;ipdb.set_trace()
+            # ------  Debug End ------
             point_cloud = DataLoadUtil.get_target_point_cloud_world_from_path(path, binocular=True)
             cam_params = DataLoadUtil.load_cam_info(path, binocular=True)
             filtered_point_cloud = ReconstructionUtil.filter_points(point_cloud, model_points_normals, cam_pose=cam_params["cam_to_world"], voxel_size=voxel_threshold, theta=filter_degree)