mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix yet ANOTHER OBO error in multi_frame_dataset

Browse Source
This commit is contained in:
James Betker 2020-11-06 20:38:34 -07:00
parent 0cf52ef52c
commit 6be6c92e5d
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
codes/data/multi_frame_dataset.py
View File

@ -19,10 +19,9 @@ class MultiFrameDataset(BaseUnsupervisedImageDataset):
frames_needed = self.num_frames frames_needed = self.num_frames
search_idx = chunk_index search_idx = chunk_index
while frames_needed > 0: while frames_needed > 0:
if source_name in self.chunk_name(search_idx): frames_needed -= 1
frames_needed -= 1 search_idx -= 1
search_idx -= 1 if source_name not in self.chunk_name(search_idx):
else:
search_idx += 1 search_idx += 1
break break