update

YingqingHe · Aug 21, 2023 · 2fb64c2 · 2fb64c2
1 parent 51bdbf6
commit 2fb64c2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lvdm/data/frame_dataset.py b/lvdm/data/frame_dataset.py
@@ -172,13 +172,13 @@ def make_dataset_ucf(dir, nframes, class_to_idx, frame_stride=1, clip_step=None)
                     "class_caption": class_caption      #boxing speed bag
                     }
                 frames.append(frame_info)
-
+            frames = frames[::frame_stride]
+
             # make videos
             if len(frames) >= nframes:
                 videos.append(frames)
 
             # make clips
-            frames = frames[::frame_stride]
             start_indices = list(range(len(frames)))[::clip_step]
             for i in start_indices:
                 clip = frames[i:i+nframes]