handle 720p with dali

OpenSportsLab · May 27, 2024 · 0c17d6d · 0c17d6d
1 parent c91d7ae
commit 0c17d6d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/.gitignore b/.gitignore
@@ -166,6 +166,8 @@ rny002_gsm
 lightning_logs
 run.sbatch
 *.out
+datasets_jsons/*
+environment.yml
 
 
 log/*

diff --git a/oslactionspotting/datasets/frame.py b/oslactionspotting/datasets/frame.py
@@ -1134,8 +1134,9 @@ def video_pipe(
             video (torch.tensor): The frames processed.
             label : the list of labels (corresponding to events) corresponding with the extracted frames.
         """
-        video, label, frame_num = fn.readers.video(
+        video, label, frame_num = fn.readers.video_resize(
             device="gpu",
+            size=(TARGET_HEIGHT,TARGET_WIDTH),
             file_list=file_list,
             sequence_length=sequence_length,
             random_shuffle=True,
@@ -1440,8 +1441,9 @@ def video_pipe(
             video (torch.tensor): The frames processed.
             label : the index of the clip in the list of clips.
         """
-        video, label = fn.readers.video(
+        video, label = fn.readers.video_resize(
             device="gpu",
+            size=(TARGET_HEIGHT,TARGET_WIDTH),
             file_list=file_list,
             sequence_length=sequence_length,
             random_shuffle=False,
@@ -1456,7 +1458,6 @@ def video_pipe(
             skip_vfr_check=True,
         )
 
-        # video = fn.resize(size=(224,398))
         video = fn.crop_mirror_normalize(
             video,
             dtype=types.FLOAT,
@@ -1468,3 +1469,5 @@ def video_pipe(
         )
 
         return video, label
+    def get_dims(video):
+        print(video.shape)