From 7cd7021bcd6c2fcfea164a01245dc43ade2bbe68 Mon Sep 17 00:00:00 2001 From: Michele Di Giorgio Date: Wed, 5 Sep 2018 16:29:28 +0100 Subject: COMPMID-1555: Fix mismatch in EqualizeHistogram function Change-Id: I1fe346a360b2d9ddd8576a1d24ff66e78c95c1f2 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/147049 Tested-by: Jenkins Reviewed-by: Anthony Barbier --- src/core/NEON/kernels/NECumulativeDistributionKernel.cpp | 2 +- src/runtime/CL/functions/CLEqualizeHistogram.cpp | 2 +- tests/validation/reference/EqualizeHistogram.cpp | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/core/NEON/kernels/NECumulativeDistributionKernel.cpp b/src/core/NEON/kernels/NECumulativeDistributionKernel.cpp index d2eac2c254..31b688c517 100644 --- a/src/core/NEON/kernels/NECumulativeDistributionKernel.cpp +++ b/src/core/NEON/kernels/NECumulativeDistributionKernel.cpp @@ -102,7 +102,7 @@ void NECumulativeDistributionKernel::run(const Window &window, const ThreadInfo } else { - const float diff = image_size - 1; + const float diff = image_size - cd_min; for(unsigned int x = 0; x < _histogram_size; ++x) { diff --git a/src/runtime/CL/functions/CLEqualizeHistogram.cpp b/src/runtime/CL/functions/CLEqualizeHistogram.cpp index 45f70d2fa7..a0663b754a 100644 --- a/src/runtime/CL/functions/CLEqualizeHistogram.cpp +++ b/src/runtime/CL/functions/CLEqualizeHistogram.cpp @@ -68,7 +68,7 @@ void calculate_cum_dist_and_lut(CLDistribution1D &dist, CLDistribution1D &cum_di } else { - const float diff = image_size - 1; + const float diff = image_size - num_lowest_pixels; for(size_t i = 0; i < 256; ++i) { diff --git a/tests/validation/reference/EqualizeHistogram.cpp b/tests/validation/reference/EqualizeHistogram.cpp index 0e966cd0b4..1a10c2c30a 100644 --- a/tests/validation/reference/EqualizeHistogram.cpp +++ b/tests/validation/reference/EqualizeHistogram.cpp @@ -66,7 +66,7 @@ SimpleTensor equalize_histogram(const SimpleTensor &src) } else { - const float diff = total_num_pixels - 1; + const float diff = total_num_pixels - cd_min; for(size_t i = 0; i < num_bins; ++i) { -- cgit v1.2.1