Function get_video_input_cases

dali/test/python/sequences_test_utils.py:424–448

(seq_layout, rng, larger_shape=(512, 288), smaller_shape=(384, 216))

def get_video_input_cases(seq_layout, rng, larger_shape=(512, 288), smaller_shape=(384, 216)):
    max_batch_size = 8
    max_num_frames = 16
    cases = []
    w, h = larger_shape
    larger = vid_source(max_batch_size, 1, max_num_frames, w, h, seq_layout)
    w, h = smaller_shape
    smaller = vid_source(max_batch_size, 2, max_num_frames, w, h, seq_layout)
    cases.append(smaller)
    samples = [sample for batch in [smaller[0], larger[0], smaller[1]] for sample in batch]
    rng.shuffle(samples)
    # test variable batch size
    case2 = [
        samples[0:1],
        samples[1 : 1 + max_batch_size],
        samples[1 + max_batch_size : 2 * max_batch_size],
        samples[2 * max_batch_size : 3 * max_batch_size],
    ]
    cases.append(case2)
    frames_idx = seq_layout.find("F")
    if frames_idx == 0:
        # test variadic number of frames in different sequences
        case3 = [[sample[: rng.randint(1, sample.shape[0])] for sample in batch] for batch in case2]
        cases.append(case3)
    return cases


@pipeline_def