Skip to content
This repository was archived by the owner on Dec 14, 2023. It is now read-only.

Commit c91771b

Browse files
Fix folder dataset random frame
1 parent 7ae53a2 commit c91771b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

utils/dataset.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -507,9 +507,9 @@ def get_frame_batch(self, vr, resize=None):
507507
effective_length = len(vr) // every_nth_frame
508508

509509
if effective_length < self.n_sample_frames:
510-
return self.__getitem__(random.randint(0, len(self.video_files) - 1))
510+
return self.__getitem__(random.randint(1, len(self.video_files) - 1))
511511

512-
effective_idx = random.randint(0, effective_length - self.n_sample_frames)
512+
effective_idx = random.randint(1, effective_length - self.n_sample_frames)
513513

514514
idxs = every_nth_frame * np.arange(effective_idx, effective_idx + self.n_sample_frames)
515515

0 commit comments

Comments
 (0)