fix:multi-fps bug. For multi-fps training, when extracting frames acc…

…ording to a certain frame_interval, the fps of the extracted frames actually changes. (#444) Co-authored-by: Jianshu Guo <[email protected]>
hpcaitech · Jun 18, 2024 · 1295446 · 1295446
1 parent 7b5791a
commit 1295446
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/opensora/datasets/datasets.py b/opensora/datasets/datasets.py
@@ -152,6 +152,8 @@ def getitem(self, index):
             # Sampling video frames
             video = temporal_random_crop(vframes, num_frames, self.frame_interval)
 
+            video_fps = video_fps // self.frame_interval
+
             # transform
             transform = get_transforms_video(self.transform_name, (height, width))
             video = transform(video)  # T C H W