upd

2024-12-31 02:52:46 +08:00
parent 2dfb6c57ce 88d44f020e
commit 9c2625b11e
9 changed files with 836 additions and 18 deletions
--- a/core/seq_dataset.py
+++ b/core/seq_dataset.py
@@ -50,11 +50,12 @@ class SeqReconstructionDataset(BaseDataset):
                if not os.path.exists(os.path.join(self.root_dir, scene_name)):
                    continue
                scene_name_list.append(scene_name)
-        return scene_name_list
+        return scene_name_list    
    
    def get_scene_name_list(self):
        return self.scene_name_list

+
    def get_datalist(self):
        datalist = []
        total = len(self.scene_name_list)