add multi seq training

2024-09-23 14:30:51 +08:00
parent 6cdff9c83f
commit 3c4077ec4f
12 changed files with 152 additions and 93 deletions
--- a/core/nbv_dataset.py
+++ b/core/nbv_dataset.py
@@ -102,9 +102,7 @@ class NBVReconstructionDataset(BaseDataset):
        max_coverage_rate = data_item_info["max_coverage_rate"]
        scene_name = data_item_info["scene_name"]
        scanned_views_pts, scanned_coverages_rate, scanned_n_to_world_pose = [], [], []
-        first_frame_idx = scanned_views[0][0]
-        first_cam_info = DataLoadUtil.load_cam_info(DataLoadUtil.get_path(self.root_dir, scene_name, first_frame_idx), binocular=True)
-        first_frame_to_world = first_cam_info["cam_to_world"]
+      
        for view in scanned_views:
            frame_idx = view[0]
            coverage_rate = view[1]