Function get_video_reader_batch

wan/data/utils.py:123–125

(video_reader, batch_index)

        gc.collect()

def get_video_reader_batch(video_reader, batch_index):
    frames = video_reader.get_batch(batch_index).asnumpy()
    return frames

def resize_frame(frame, target_short_side):
    h, w, _ = frame.shape