Merge pull request #17 from zyj-2000/main

Fix the BUG: Video Quality Scorer -> Python API: Missing the "load_video"

q_align/__init__.py

@@ -1,2 +1,3 @@
 from .model import MPLUGOwl2LlamaForCausalLM
-from .evaluate import QAlignScorer, QAlignAestheticScorer, QAlignVideoScorer
+from .evaluate import QAlignScorer, QAlignAestheticScorer, QAlignVideoScorer
+from .load_video import load_video

q_align/load_video.py

@@ -0,0 +1,13 @@
+from PIL import Image
+
+def load_video(video_file):
+    from decord import VideoReader
+    vr = VideoReader(video_file)
+
+    # Get video frame rate
+    fps = vr.get_avg_fps()
+
+    # Calculate frame indices for 1fps
+    frame_indices = [int(fps * i) for i in range(int(len(vr) / fps))]
+    frames = vr.get_batch(frame_indices).asnumpy()
+    return [Image.fromarray(frames[i]) for i in range(int(len(vr) / fps))]