Function compare_videos

dali/test/python/decoder/test_video.py:166–178

(reference, actual)

def compare_videos(reference, actual):
    try:
        np.testing.assert_array_equal(reference, actual)
    except AssertionError:
        # Save frames as PNG files for debugging
        print("Videos don't match - saving frames as PNG files for debugging...")
        for i, (ref_frame, act_frame) in enumerate(zip(reference, actual)):
            ref_filename = f"reference_frame_{i}.png"
            act_filename = f"actual_frame_{i}.png"
            cv2.imwrite(ref_filename, ref_frame)
            cv2.imwrite(act_filename, act_frame)
            print(f"Saved frame {i} to {ref_filename} and {act_filename}")
        raise


@pipeline_def(device_id=0)