aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/core/NEON/kernels/NECumulativeDistributionKernel.cpp2
-rw-r--r--src/runtime/CL/functions/CLEqualizeHistogram.cpp2
-rw-r--r--tests/validation/reference/EqualizeHistogram.cpp2
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<T> equalize_histogram(const SimpleTensor<T> &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)
{