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 个通道，即它必须是灰色的，而不是彩色的。您在roi 上调用countNonZero，这是binaryImage 的子图像，它是src 的克隆，最初是彩色的。

我想你想写cvtColor(binaryImage, binaryImage, CV_BGR2GRAY);。在这种情况下，这是有道理的。但是，我没有看到您再次在任何地方使用src，因此您可能不需要此中间图像。如果您这样做，请不要称“二进制”，因为计算机视觉中的“二进制”通常代表黑白图像，只有两种颜色。您的图像是“灰色”的，因为它具有各种黑白色调。

关于你最初的任务，Miki 是对的，你应该使用cv::reduce。他已经给你举例说明了如何使用它。

Answer 2

顺便说一句，您可以使用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);