Merge pull request #3611 from jet47:fix-gpumat-setto

10 years ago · e501c61c04
parent aacb9b7cf5 f676bfb3d7
commit e501c61c04
1 changed files with 5 additions and 0 deletions
--- a/modules/core/src/cuda/gpu_mat.cu
+++ b/modules/core/src/cuda/gpu_mat.cu
@ -390,6 +390,11 @@ GpuMat& cv::cuda::GpuMat::setTo(Scalar value, InputArray _mask, Stream& stream)

    GpuMat mask = _mask.getGpuMat();

+    if (mask.empty())
+    {
+        return setTo(value, stream);
+    }
+
    CV_DbgAssert( size() == mask.size() && mask.type() == CV_8UC1 );

    typedef void (*func_t)(const GpuMat& mat, const GpuMat& mask, Scalar scalar, Stream& stream);