c++ - OpenCV - 随机森林示例

Question

有没有人有一些使用带有 2.3.1 API Mat 而不是 cvMat 的随机森林的例子？

基本上我有一个矩阵垫数据，它由 1000 行和 16x16x3 元素组成，矩阵垫响应一个 1000x1 矩阵，其中包含每行所属的类。我想对此运行随机森林算法。

Answer 1

您已经获得了正确格式的数据；剩下的就是实例化一个 CvRTrees 对象并执行您的预测。

可以在此处找到 Random Trees v2.3 的文档。您还需要查看CvStatModel::train()文档，该文档实际上包含CvRTree::train. Tom 在您应该使用的评论中引用了一个很好的完整示例。

除了您的数据，您还需要一个 Mat 来指定每个属性的类型。此 Mat 每个输入属性都有一行，输出类型有另外一行（在您的情况下为 16x16x3 + 1 行）。

或者，您可以使用 CvRTParams 对象来指定诸如树数、最大深度等参数。我在下面的示例中使用默认值。

如果您愿意，您可以传入 valIdx 和 sampleIdx Mats，它们分别指定哪些属性和哪些数据行用于训练。这对于选择训练/验证数据很有用，而无需做一堆体操来将它们放在单独的 Mats 中。

这是一个简单的例子：

#define ATTRIBUTES_PER_SAMPLE (16*16*3)
// Assumes training data (1000, 16x16x3) are in training_data
// Assumes training classifications (1000, 1) are in training_classifications

// All inputs are numerical. You can change this to reflect your data
Mat var_type = Mat(ATTRIBUTES_PER_SAMPLE + 1, 1, CV_8U );
var_type.setTo(Scalar(CV_VAR_NUMERICAL) ); // all inputs are numerical

// Output is a category; this is classification, not regression
var_type.at<uchar>(ATTRIBUTES_PER_SAMPLE, 0) = CV_VAR_CATEGORICAL;

// Train the classifier
CvRTrees* rtree = new CvRTrees;
rtree->train(training_data, CV_ROW_SAMPLE, training_classifications,
             Mat(), Mat(), var_type);