opencv
diff --git a/‎modules/imgproc/include/opencv2/imgproc.hpp
Copy file name to clipboard
+20-1Lines changed: 20 additions & 1 deletion b/‎modules/imgproc/include/opencv2/imgproc.hpp
Copy file name to clipboard
+20-1Lines changed: 20 additions & 1 deletion
diff --git a/‎modules/imgproc/src/opencl/threshold_mask.cl
Copy file name to clipboard
+60Lines changed: 60 additions & 0 deletions b/‎modules/imgproc/src/opencl/threshold_mask.cl
Copy file name to clipboard
+60Lines changed: 60 additions & 0 deletions
@@ -3081,11 +3081,30 @@ types.
 @param type thresholding type (see #ThresholdTypes).
 @return the computed threshold value if Otsu's or Triangle methods used.
 
-@sa  adaptiveThreshold, findContours, compare, min, max
+@sa  thresholdWithMask, adaptiveThreshold, findContours, compare, min, max
  */
 CV_EXPORTS_W double threshold( InputArray src, OutputArray dst,
                                double thresh, double maxval, int type );
 
+/** @brief Same as #threshold, but with an optional mask
+
+@note If the mask is empty, #thresholdWithMask is equivalent to #threshold.
+If the mask is not empty, dst *must* be of the same size and type as src, so that
+outliers pixels are left as-is
+
+@param src input array (multiple-channel, 8-bit or 32-bit floating point).
+@param dst output array of the same size  and type and the same number of channels as src.
+@param mask optional mask (same size as src, 8-bit).
+@param thresh threshold value.
+@param maxval maximum value to use with the #THRESH_BINARY and #THRESH_BINARY_INV thresholding
+types.
+@param type thresholding type (see #ThresholdTypes).
+@return the computed threshold value if Otsu's or Triangle methods used.
+
+@sa  threshold, adaptiveThreshold, findContours, compare, min, max
+*/
+CV_EXPORTS_W double thresholdWithMask( InputArray src, InputOutputArray dst, InputArray mask,
+                                       double thresh, double maxval, int type );
 
 /** @brief Applies an adaptive threshold to an array.
 
 
@@ -0,0 +1,60 @@
+// This file is part of OpenCV project.
+// It is subject to the license terms in the LICENSE file found in the top-level directory
+// of this distribution and at https://opencv.org/license.html.
+// @Authors
+//    Zhang Ying, zhangying913@gmail.com
+//    Pierre Chatelier, pierre@chachatelier.fr
+
+#ifdef DOUBLE_SUPPORT
+#ifdef cl_amd_fp64
+#pragma OPENCL EXTENSION cl_amd_fp64:enable
+#elif defined (cl_khr_fp64)
+#pragma OPENCL EXTENSION cl_khr_fp64:enable
+#endif
+#endif
+
+__kernel void threshold_mask(__global const uchar * srcptr, int src_step, int src_offset,
+                             __global uchar * dstptr, int dst_step, int dst_offset, int rows, int cols,
+                             __global const uchar * maskptr, int mask_step, int mask_offset,
+                             T1 thresh, T1 max_val, T1 min_val)
+{
+    int gx = get_global_id(0);
+    int gy = get_global_id(1) * STRIDE_SIZE;
+
+    if (gx < cols)
+    {
+        int src_index = mad24(gy, src_step, mad24(gx, (int)sizeof(T), src_offset));
+        int dst_index = mad24(gy, dst_step, mad24(gx, (int)sizeof(T), dst_offset));
+        int mask_index = mad24(gy, mask_step, mad24(gx/CN, (int)sizeof(uchar), mask_offset));
+
+        #pragma unroll
+        for (int i = 0; i < STRIDE_SIZE; i++)
+        {
+            if (gy < rows)
+            {
+                T sdata = *(__global const T *)(srcptr + src_index);
+                const uchar mdata = *(maskptr + mask_index);
+                if (mdata != 0)
+                {
+                    __global T * dst = (__global T *)(dstptr + dst_index);
+
+                    #ifdef THRESH_BINARY
+                            dst[0] = sdata > (thresh) ? (T)(max_val) : (T)(0);
+                    #elif defined THRESH_BINARY_INV
+                            dst[0] = sdata > (thresh) ? (T)(0) : (T)(max_val);
+                    #elif defined THRESH_TRUNC
+                            dst[0] = clamp(sdata, (T)min_val, (T)(thresh));
+                    #elif defined THRESH_TOZERO
+                            dst[0] = sdata > (thresh) ? sdata : (T)(0);
+                    #elif defined THRESH_TOZERO_INV
+                            dst[0] = sdata > (thresh) ? (T)(0) : sdata;
+                    #endif
+                }
+                gy++;
+                src_index += src_step;
+                dst_index += dst_step;
+                mask_index += mask_step;
+            }
+        }
+    }
+}