global: debug inference

2024-11-01 22:51:16 +00:00
parent 982a3b9b60
commit 287983277a
3 changed files with 10 additions and 2 deletions
--- a/core/seq_dataset.py
+++ b/core/seq_dataset.py
@@ -49,6 +49,9 @@ class SeqReconstructionDataset(BaseDataset):
                scene_name = line.strip()
                scene_name_list.append(scene_name)
        return scene_name_list
+    
+    def get_scene_name_list(self):
+        return self.scene_name_list

    def get_datalist(self):
        datalist = []