Функция countNonZero дает ошибку утверждения в openCV

Question

Я попытался получить горизонтальную проекцию, используя функцию countNonZero(), как показано ниже.

Mat src = imread(INPUT_FILE, CV_LOAD_IMAGE_COLOR); 
Mat binaryImage = src.clone(); 
cvtColor(src, src, CV_BGR2GRAY); 

Mat horizontal = Mat::zeros(1,binaryImage.cols, CV_8UC1); 

for (int i = 0; i<binaryImage.cols; i++) 
{ 
    Mat roi = binaryImage(Rect(0, 0, 1, binaryImage.rows)); 

    horizontal.at<int>(0,i) = countNonZero(roi); 
    cout << "Col no:" << i << " >>" << horizontal.at<int>(0, i); 
} 

Но ошибка произошла в строке вызова функции countonZero(). Ошибка заключается в следующем.

OpenCV Error: Assertion failed (src.channels() == 1 && func != 0) in cv::countNo 
    nZero, file C:\builds\2_4_PackSlave-win32-vc12-shared\opencv\modules\core\src\st 
    at.cpp, line 549 

Может кто-нибудь указать на ошибку?

Answer 1

утверждения src.channels() == 1 означает, что изображение должно иметь 1 канал, то есть он должен быть серым, а не цвета. Вы звоните countNonZero по адресу roi, который представляет собой прообраз binaryImage, который является клоном src, который изначально окрашен.

Я полагаю, вы хотели написать cvtColor(binaryImage, binaryImage, CV_BGR2GRAY);. В этом случае это имеет смысл. Однако я не вижу, чтобы вы снова использовали src, поэтому, возможно, вам не понадобится это промежуточное изображение. В случае, если вы это делаете, не называйте «двоичный», так как «двоичный» в компьютерном зрении обычно обозначает черно-белое изображение, только два цвета. Ваше изображение «серое», так как оно имеет все оттенки черно-белого.

Что касается вашей первоначальной задачи, Мики прав, вы должны использовать для этого cv::reduce. Он уже дал вам пример о том, как его использовать.

Answer 2

BTW, вы можете вычислить горизонтальную проекцию, используя reduce в качестве аргумента CV_REDUCE_SUM.

Минимальный пример:

Mat1b mat(4, 4, uchar(0)); 
mat(0,0) = uchar(1); 
mat(0,1) = uchar(1); 
mat(1,1) = uchar(1); 

// mat is: 
// 
// 1100 
// 0100 
// 0000 
// 0000 

// Horizontal projection, result would be a column matrix 
Mat1i reducedHor; 
cv::reduce(mat, reducedHor, 1, CV_REDUCE_SUM); 

// reducedHor is: 
// 
// 2 
// 1 
// 0 
// 0 

// Vertical projection, result would be a row matrix 
Mat1i reducedVer; 
cv::reduce(mat, reducedVer, 0, CV_REDUCE_SUM); 

// reducedVer is: 
// 
// 1200 


// Summary 
// 
// 1100 > 2 
// 0100 > 1 
// 0000 > 0 
// 0000 > 0 
// 
// vvvv 
// 1200 

Вы можете использовать это с вашими изображениями, как это:

// RGB image 
Mat3b img = imread("path_to_image"); 

// Gray image, contains values in [0,255] 
Mat1b gray; 
cvtColor(img, gray, CV_BGR2GRAY); 

// Binary image, contains only 0,1 values 
// The sum of pixel values will equal the count of non-zero pixels 
Mat1b binary; 
threshold(gray, binary, 1, 1, THRESH_BINARY); 

// Horizontal projection 
Mat1i reducedHor; 
cv::reduce(binary, reducedHor, 1, CV_REDUCE_SUM); 

// Vertical projection 
Mat1i reducedVer; 
cv::reduce(binary, reducedVer, 0, CV_REDUCE_SUM);