storytold/FineTrainers-Conditioning
2
0
Fork 0
mirror of https://github.com/storytold/FineTrainers-Conditioning.git synced 2026-10-18 07:33:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

updates

Browse Source
This commit is contained in:
sayakpaul
2024-11-18 15:30:23 +05:30
parent a40ccd2237
commit 4c05ea6669
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
training/dataset.py
-2
View File
@@ -342,8 +342,6 @@ class VideoDatasetWithResizeAndRectangleCrop(VideoDataset):
nearest_frame_bucket = min( nearest_frame_bucket = min(
self.frame_buckets, key=lambda x: abs(x - min(video_num_frames, self.max_num_frames)) self.frame_buckets, key=lambda x: abs(x - min(video_num_frames, self.max_num_frames))
) )
if video_num_frames < nearest_frame_bucket:
return None, None, None
frame_indices = list(range(0, video_num_frames, video_num_frames // nearest_frame_bucket)) frame_indices = list(range(0, video_num_frames, video_num_frames // nearest_frame_bucket))