Skip to content

Commit

Permalink
Merge pull request #5 from eltociear/patch-1
Browse files Browse the repository at this point in the history
Update video_utils.py
  • Loading branch information
csuhan authored Dec 10, 2023
2 parents 33df818 + fd94129 commit eb0c0f9
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions data/video_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ def uniform_crop(images, size, spatial_idx, boxes=None, scale_size=None):
crop if height is larger than width.
boxes (ndarray or None): optional. Corresponding boxes to images.
Dimension is `num boxes` x 4.
scale_size (int): optinal. If not None, resize the images to scale_size before
scale_size (int): optional. If not None, resize the images to scale_size before
performing any crop.
Returns:
cropped (tensor): images with dimension of
Expand Down Expand Up @@ -201,4 +201,4 @@ def load_and_transform_video_data(
if __name__ == '__main__':
video_path = "datasets/InstructionTuning/video/music_aqa/MUSIC-AVQA-videos-Real/00000002.mp4"
video, audio = load_and_transform_video_data(video_path, video_path, clip_duration=1, clips_per_video=5, with_audio=True)
import pdb;pdb.set_trace()
import pdb;pdb.set_trace()

0 comments on commit eb0c0f9

Please sign in to comment.