Function vid_source

dali/test/python/sequences_test_utils.py:473–488

(batch_size, num_batches, num_frames, width, height, seq_layout)

def vid_source(batch_size, num_batches, num_frames, width, height, seq_layout):
    pipe = vid_pipeline(
        num_threads=4,
        batch_size=batch_size,
        num_frames=num_frames,
        width=width,
        height=height,
        device_id=0,
        seq_layout=seq_layout,
        prefetch_queue_depth=1,
    )
    batches = []
    for _ in range(num_batches):
        (pipe_out,) = pipe.run()
        batches.append(as_batch(pipe_out))
    return batches


def video_suite_helper(ops_test_cases, test_channel_first=True, expand_channels=False, rng=None):