4

我对 OpenCV 很陌生。我正在尝试使用 CvNormalBayesClassifier 来训练我的程序来学习皮肤像素颜色。

目前，我在不同的光照条件和背景下获得了大约 20 张人体照片（面部/其他身体部位）。我也得到了 20 个相应的响应，其中皮肤部分标记为红色，其他部分标记为绿色。

我无法理解如何使用该功能

bool CvNormalBayesClassifier::train(const CvMat* _train_data, const CvMat* _response, const Cv*Mat _var_idx = 0, const CvMat* _sample_idx=0,, bool update=false);

我应该如何使用我必须准备可以作为_train_data和_responses传入的值的当前两个图片库？

非常感谢。

4

2 回答 2

4

您需要将训练图像中的像素值放入 train_data 中，并在响应中输入与该像素的类别相对应的索引（例如，1 表示皮肤类，0 表示非皮肤类）。var_idx 和 sample_idx 可以保持原样，它们用于屏蔽训练集中的一些描述符或样本。根据您是否一次获得所有描述符（所有训练图像的所有像素）将更新设置为真/假，以防您可以让它为假，或者您逐步处理训练图像（这可能更好地解决内存问题)，在这种情况下，您需要更新模型。

让我用代码澄清一下（未检查，并使用我强烈推荐的 OpenCV 的 C++ 接口而不是旧的 C）

int main(int argc, char **argv)
{

  CvNormalBaseClassifier classifier;
  for (int i = 0; i < argc; ++i) {
    cv::Mat image = // read in your training image, say cv::imread(argv[i]);
    // read your mask image
    cv::Mat mask = ...
    cv::Mat response = mask == CV_RGB(255,0,0); // little trick: you said red pixels in your mask correspond to skin, so pixels in responses are set to 1 if corresponding pixel in mask is red, 0 otherwise.
    cv::Mat responseInt;
    response.convertTo(responsesInt, CV_32S); // train expects a matrix of integers

    image = image.reshape(0, image.rows*image.cols); // little trick number 2 convert your width x height, N channel image into a witdth*height row matrix by N columns, as each pixel should be considere as a training sample.
    responsesInt = responsesInt.reshape(0, image.rows*image.cols); // the same, image and responses have the same number of rows (w*h).

    classifier.train(image, responsesInt, 0, 0, true);

}

于 2012-11-06T22:27:15.333 回答
0

我对这个类进行了谷歌搜索，但没有找到太多信息，实际上即使是官方的opencv文档也没有提供参数的直接说明。但我确实注意到 opencv 文档中的一件事

该方法训练 Normal Bayes 分类器。它遵循通用 CvStatModel::train() 方法的约定，但有以下限制：

它引导我进入 CvStatModel 类，从那里我发现了一些有用的东西。也许你也可以从第 471 页看这本书，它为你提供了更多关于这门课的细节。这本书是免费的谷歌图书。

于 2012-11-06T16:31:09.770 回答