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

nearest_frame_bucket.

Browse Source
This commit is contained in:
sayakpaul
2024-11-19 10:27:24 +05:30
parent 1409d478d5
commit c80bd177fd
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
training/dataset.py
+6 -2
View File
@@ -277,7 +277,9 @@ class VideoDatasetWithResizing(VideoDataset):
video_reader = decord.VideoReader(uri=path.as_posix())
video_num_frames = len(video_reader)
nearest_frame_bucket = min(
self.frame_buckets, key=lambda x: abs(x - min(video_num_frames, self.max_num_frames))
[bucket for bucket in self.frame_buckets if bucket <= video_num_frames],
key=lambda x: abs(x - min(video_num_frames, self.max_num_frames)),
default=1,
)
frame_indices = list(range(0, video_num_frames, video_num_frames // nearest_frame_bucket))
@@ -342,7 +344,9 @@ class VideoDatasetWithResizeAndRectangleCrop(VideoDataset):
video_reader = decord.VideoReader(uri=path.as_posix())
video_num_frames = len(video_reader)
nearest_frame_bucket = min(
self.frame_buckets, key=lambda x: abs(x - min(video_num_frames, self.max_num_frames))
[bucket for bucket in self.frame_buckets if bucket <= video_num_frames],
key=lambda x: abs(x - min(video_num_frames, self.max_num_frames)),
default=1,
)
frame_indices = list(range(0, video_num_frames, video_num_frames // nearest_frame_bucket))