From 07eef7383815dc3e1556b8bccd1d7fff56cf4f3b Mon Sep 17 00:00:00 2001 From: Anup Parikh Date: Thu, 18 Jul 2019 10:31:50 -0600 Subject: [PATCH] Ignore NaN values in bilateral filter --- modules/cudaimgproc/src/cuda/bilateral_filter.cu | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/modules/cudaimgproc/src/cuda/bilateral_filter.cu b/modules/cudaimgproc/src/cuda/bilateral_filter.cu index f81adc7a9..c077a833f 100644 --- a/modules/cudaimgproc/src/cuda/bilateral_filter.cu +++ b/modules/cudaimgproc/src/cuda/bilateral_filter.cu @@ -98,6 +98,8 @@ namespace cv { namespace cuda { namespace device continue; value_type value = saturate_cast(src(cy, cx)); + if (isnan(norm_l1(value))) + continue; float weight = ::exp(space2 * sigma_spatial2_inv_half + sqr(norm_l1(value - center)) * sigma_color2_inv_half); sum1 = sum1 + weight * value; @@ -114,6 +116,8 @@ namespace cv { namespace cuda { namespace device continue; value_type value = saturate_cast(b.at(cy, cx, src.data, src.step)); + if (isnan(norm_l1(value))) + continue; float weight = ::exp(space2 * sigma_spatial2_inv_half + sqr(norm_l1(value - center)) * sigma_color2_inv_half);