inference on YCB

2024-12-04 14:52:23 +08:00
parent 501975457f
commit 47624f12cf
9 changed files with 25 additions and 18 deletions
--- a/core/seq_dataset_preprocessed.py
+++ b/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):