如何在OpenCV随机树中设置树的数量?

时间:2015-10-10 00:38:38

标签: c++ random-forest opencv3.0

我是OpenCV的新手。用于使用R中的随机森林库,它允许您设置要训练的树的数量。是否可以在OpenCV 3.0中设置此值?

此外,还可以调整哪些其他参数?

这就是我现在所拥有的:

 // randome forest classifier
Ptr<RTrees> rfClassifier = RTrees::create();
rfClassifier->setMaxDepth(5);
rfClassifier->setMinSampleCount(3);
rfClassifier->setTermCriteria(TermCriteria(TermCriteria::MAX_ITER, 100, 1e-6));

我检查了TermCriteria :: Type,似乎我应该将它设置为使用这样的50棵树?

int numTrees = 50;    
TermCriteria(TermCriteria::COUNT, numTrees, 1e-6)

1 个答案:

答案 0 :(得分:0)

是的,你是对的。

训练随机森林模型后,您可以保存文件并检查模型文件中指定的树的数量。

在模型文件中,您可以找到名为ntrees:

的字段