inference on YCB

core/seq_dataset.py

@@ -179,9 +179,9 @@ if __name__ == "__main__":
     np.random.seed(seed)
     
     config = {
-        "root_dir": "/media/hofee/data/data/new_testset",
+        "root_dir": "/media/hofee/data/results/ycb_view_data",
         "source": "seq_reconstruction_dataset",
-        "split_file": "/media/hofee/data/data/OmniObject3d_test.txt",
+        "split_file": "/media/hofee/data/results/ycb_test.txt",
         "load_from_preprocess": True,
         "filter_degree": 75,
         "num_workers": 0,
@@ -189,7 +189,7 @@ if __name__ == "__main__":
         "type": namespace.Mode.TEST,
     }
     
-    output_dir = "/media/hofee/data/data/new_testset_output"
+    output_dir = "/media/hofee/data/results/ycb_preprocessed_dataset"
     os.makedirs(output_dir, exist_ok=True)
 
     ds = SeqReconstructionDataset(config)

core/seq_dataset_preprocessed.py

@@ -21,7 +21,7 @@ class SeqReconstructionDatasetPreprocessed(BaseDataset):
         super(SeqReconstructionDatasetPreprocessed, self).__init__(config)
         self.config = config
         self.root_dir = config["root_dir"]
-        self.real_root_dir = r"/media/hofee/data/data/new_testset"
+        self.real_root_dir = r"/media/hofee/data/results/ycb_view_data"
         self.item_list = os.listdir(self.root_dir)
 
     def __getitem__(self, index):