更新OpenCV CvNormalBayesClassifier

Question

我正在尝试使用CvNormalBayesClassifier来训练我的程序以学习皮肤像素颜色。我有一套训练图像和响应图像。响应图像为黑色和白色，皮肤区域标记为白色。以下是我的代码，

CvNormalBayesClassifier classifier;
for (int i = 0; i < numFiles; i++) {

    string trainFile = "images/" + int2str(i) + ".jpg";
    string responseFile = "images/" + int2str(i) + "_mask.jpg";

    Mat trainData = imread(trainFile, 1);
    Mat responseData = imread(responseFile, CV_LOAD_IMAGE_GRAYSCALE);

    trainData = trainData.reshape(1, trainData.rows * trainData.cols);
    responseData = responseData.reshape(0, responseData.rows * responseData.cols);

    trainData.convertTo(trainData, CV_32FC1);
    responseData.convertTo(responseData, CV_32FC1);

    classifier.train(trainData, responseData, Mat(), Mat(), i != 0);
}

但是，它会出现以下错误，

The function/feature is not implemented (In the current implementation the new training data must have absolutely the same set of class labels as used in the original training data) in CvNormalBayesClassifier::train

非常感谢。

Answer 1

如错误消息所述，您无法根据新的类标签“更新”分类器。普通贝叶斯分类器学习高斯混合物来表示训练数据。如果您突然开始添加新标签，这种混合模型将不再正确，必须从头开始学习新模型。

Answer 2

好吧，我发现问题是黑白图像已被压缩，因此包含0到255之间的值。因此，其他图像中可能会有新的类标签。

要解决此问题，请使用阈值使值all为0或255。