From f7fd5b0490d275052011f56427a167cc566310ff Mon Sep 17 00:00:00 2001 From: Wang Hangjie Date: Thu, 24 Oct 2024 09:54:34 -0700 Subject: [PATCH] Enhance PSNR Check for Luma (Y) Channel in VPP Sharpen Filter The current implementation of the PSNR check for the Luma (Y) channel raises exceptions for actual values exceeding reference values. This behavior is not reflective of typical processing scenarios where higher actual values are expected. Updated the `compare` function to only raise exceptions if the actual Luma (Y) value is lower than the reference value. Signed-off-by: Wang Hangjie --- lib/mixin/vpp.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/lib/mixin/vpp.py b/lib/mixin/vpp.py index bc3cce8b..40191be4 100644 --- a/lib/mixin/vpp.py +++ b/lib/mixin/vpp.py @@ -132,7 +132,9 @@ def compare(k, ref, actual): assert actual[-2] == 100, "Cb(U) should not be affected by SHARPEN filter" assert actual[-1] == 100, "Cr(V) should not be affected by SHARPEN filter" assert ref is not None, "Invalid reference value" - assert abs(ref[-3] - actual[-3]) < 0.25, "Luma (Y) out of baseline range" + # Check if the actual Luma (Y) value is significantly lower than the reference value. + # If the actual value is lower than the reference by more than 0.25, raise an exception. + assert ref[-3] - actual[-3] < 0.25, "Luma (Y) out of baseline range" metrics2.check( metric = dict(type = "psnr"), compare = compare,