Fix OBO error in multi_frame_dataset
In some datasets, this meant one frame was included in a sequence where it didn't belong. In datasets with mismatched chunk sizes, this resulted in an error.
This commit is contained in:
parent
e990be0449
commit
c21088e238
|
@ -15,9 +15,9 @@ class MultiFrameDataset(BaseUnsupervisedImageDataset):
|
||||||
def get_sequential_image_paths_from(self, chunk_index, chunk_offset):
|
def get_sequential_image_paths_from(self, chunk_index, chunk_offset):
|
||||||
im_name = self.chunk_name(chunk_index)
|
im_name = self.chunk_name(chunk_index)
|
||||||
source_name = im_name[:-12]
|
source_name = im_name[:-12]
|
||||||
frames_needed = self.num_frames - 1
|
|
||||||
# Search backwards for the frames needed. We are assuming that every video in the dataset has at least frames_needed frames.
|
# Search backwards for the frames needed. We are assuming that every video in the dataset has at least frames_needed frames.
|
||||||
search_idx = chunk_index-1
|
frames_needed = self.num_frames
|
||||||
|
search_idx = chunk_index
|
||||||
while frames_needed > 0:
|
while frames_needed > 0:
|
||||||
if source_name in self.chunk_name(search_idx):
|
if source_name in self.chunk_name(search_idx):
|
||||||
frames_needed -= 1
|
frames_needed -= 1
|
||||||
|
|
Loading…
Reference in New Issue
Block a user