generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

accept multiple channels image for threshold on cuda

Browse Source 
* modify test, too
This commit is contained in:
Tomoaki Teshima
2016-05-16 06:45:35 +09:00
parent d6b8563530
commit b6a093fba1
2 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules/cudaarithm/src/cuda/threshold.cu
View File

@@ -101,10 +101,12 @@ double cv::cuda::threshold(InputArray _src, OutputArray _dst, double thresh, dou
const int depth = src.depth();
CV_Assert( src.channels() == 1 && depth <= CV_64F );
CV_Assert( depth <= CV_64F );
CV_Assert( type <= 4 /*THRESH_TOZERO_INV*/ );
GpuMat dst = getOutputMat(_dst, src.size(), src.type(), stream);
src = src.reshape(1);
dst = dst.reshape(1);
if (depth == CV_32F && type == 2 /*THRESH_TRUNC*/)
{