Merge pull request #198 from ShowLo/patch-2

Fix the bug that causes an infinite loop when the total number of fra…
2026-02-04 09:29:20 +08:00 · 2024-09-23 13:28:35 +08:00
parent b3c30c3561 4f69a9cfdd
commit 4724ed8a78
1 changed files with 1 additions and 1 deletions
--- a/train_codes/DataLoader.py
+++ b/train_codes/DataLoader.py
@@ -135,7 +135,7 @@ class Dataset(object):
            #随机选择某个video里
            vidname = self.all_videos[idx].split('/')[-1]
            video_imgs = self.all_img_names[idx]
-            if len(video_imgs) == 0:
+            if len(video_imgs) <= 11:
                continue
            img_name = random.choice(video_imgs)
            img_idx = int(basename(img_name).split(".")[0])