Function run_video_decoding_test

dali/test/python/test_dali_video_plugin_decoder.py:39–57

(test_file_path, frame_list_file_path, frames_directory_path)

def run_video_decoding_test(test_file_path, frame_list_file_path, frames_directory_path):
    pipeline = video_decoder_pipeline(test_file_path)

    (output,) = pipeline.run()
    frames = output.as_cpu().as_array()

    with open(frame_list_file_path) as f:
        frame_list = f.read().splitlines()

    for i, frame in enumerate(frames[0]):
        # Check if the frame is equal to the ground truth frame.
        # Due to differences in how the decoding is implemented in
        # different video codecs, we can't guarantee that the frames
        # will be exactly the same. Main purpose of this test is to
        # check if the decoding is working and we hit the correct frames.
        ground_truth = cv2.imread(f"{frames_directory_path}/{frame_list[i]}")
        frame = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR)
        if not np.average(frame - ground_truth) < 15:
            assert False, f"Frame {i} is not equal"


def test_cfr_h264_mp4_decoding():