Skip to content

Commit

Permalink
Update dataloader
Browse files Browse the repository at this point in the history
  • Loading branch information
YunzeMan authored Oct 24, 2024
1 parent 9d11d09 commit 6568a2c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion scenecraft/data/dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -1158,7 +1158,7 @@ def __init__(self, raw_root_dir: str='', root_dir: str='./data/hypersim', **kwar
all_scenes = list(filter(lambda x: x.startswith('ai'), all_scenes))
with open(SPLIT_PATHS['hypersim']['train'], 'r') as file:
avail_scene = [line.strip() for line in file.readlines()]
self.split_available_scenes = list(filter(lambda x: x in avail_scene, all_scenes)).sort()
self.split_available_scenes = sorted(filter(lambda x: x in avail_scene, all_scenes))
if self.scene_ids:
assert set(self.scene_ids).issubset(set(self.split_available_scenes)), f"Given scene {self.scene_ids} not entirely exist in splits {split}."
self.split_available_scenes = list(filter(lambda x: x in self.scene_ids, self.split_available_scenes))
Expand Down

0 comments on commit 6568a2c

Please sign in to comment.