upd infernce

core/seq_dataset.py

@@ -8,7 +8,7 @@ import torch
 import os
 import sys
 
-sys.path.append(r"/data/hofee/project/nbv_rec/nbv_reconstruction")
+sys.path.append(r"C:\Document\Local Project\nbv_rec\nbv_reconstruction")
 
 from utils.data_load import DataLoadUtil
 from utils.pose import PoseUtil
@@ -55,7 +55,9 @@ class SeqReconstructionDataset(BaseDataset):
 
     def get_datalist(self):
         datalist = []
-        for scene_name in self.scene_name_list:
+        total = len(self.scene_name_list)
+        for idx, scene_name in enumerate(self.scene_name_list):
+            print(f"processing {scene_name} ({idx}/{total})")
             seq_num = DataLoadUtil.get_label_num(self.root_dir, scene_name)
             scene_max_coverage_rate = 0
             max_coverage_rate_list = []
@@ -182,17 +184,33 @@ if __name__ == "__main__":
     seed = 0
     torch.manual_seed(seed)
     np.random.seed(seed)
+    '''
+    OmniObject3d_test:
+    root_dir: "H:\\AI\\Datasets\\packed_test_data"
+    model_dir: "H:\\AI\\Datasets\\scaled_object_meshes"
+    source: seq_reconstruction_dataset
+    split_file: "H:\\AI\\Datasets\\data_list\\OmniObject3d_test.txt"
+    type: test
+    filter_degree: 75
+    eval_list:
+      - pose_diff
+      - coverage_rate_increase
+    ratio: 0.1
+    batch_size: 1
+    num_workers: 12
+    pts_num: 8192
+    load_from_preprocess: True
+    '''
     config = {
-        "root_dir": "/data/hofee/data/new_full_data",
+        "root_dir": "H:\\AI\\Datasets\\packed_test_data",
         "source": "seq_reconstruction_dataset",
-        "split_file": "/data/hofee/data/sample.txt",
+        "split_file": "H:\\AI\\Datasets\\data_list\\OmniObject3d_test.txt",
         "load_from_preprocess": True,
-        "ratio": 0.5,
-        "batch_size": 2,
+        "ratio": 1,
         "filter_degree": 75,
         "num_workers": 0,
-        "pts_num": 4096,
-        "type": namespace.Mode.TRAIN,
+        "pts_num": 8192,
+        "type": "test",
     }
     ds = SeqReconstructionDataset(config)
     print(len(ds))