Merge remote-tracking branch 'origin/gan_lab' into gan_lab
This commit is contained in:
commit
85c545835c
|
@ -29,6 +29,9 @@ class MultiFrameDataset(BaseUnsupervisedImageDataset):
|
||||||
# Now build num_frames starting from search_idx.
|
# Now build num_frames starting from search_idx.
|
||||||
hqs, refs, masks, centers = [], [], [], []
|
hqs, refs, masks, centers = [], [], [], []
|
||||||
for i in range(self.num_frames):
|
for i in range(self.num_frames):
|
||||||
|
idx = search_idx + i
|
||||||
|
if idx < 0 or idx >= len(self.chunks) or chunk_offset < 0 or chunk_offset >= len(self.chunks[idx]):
|
||||||
|
print("Chunk reference indexing failed for %s." % (im_name,), search_idx, i, chunk_offset, self.num_frames)
|
||||||
h, r, c, m, p = self.chunks[search_idx + i][chunk_offset]
|
h, r, c, m, p = self.chunks[search_idx + i][chunk_offset]
|
||||||
hqs.append(h)
|
hqs.append(h)
|
||||||
refs.append(r)
|
refs.append(r)
|
||||||
|
|
Loading…
Reference in New Issue
Block a user