How to load trained RTrees?

[lymhust]

Hi,

I have trained a RTrees model and saved it using following codes. However, I cannot reload the trained model. How to solve this problem? Thanks!

local rt = cv.ml.RTrees()
........
rt:save('./RT_MNIST_MODEL')

How to reload RT_MNIST_MODEL?

[lymhust]

I have tried to wrap the OpenCV function: cv::Ptr<ml::SVM> p = ml::SVM::load<ml::SVM>(filename) using the following codes and solved the problem. Unfortunately, OpenCV has a bug in this function.....

In ml.cpp:

// SVM
extern "C"
struct SVMPtr SVM_load(const char *filename) {
  return rescueObjectFromPtr(ml::SVM::load<ml::SVM>(filename));
}

In ml/init.lua:

function SVM:load(filename)
   self.ptr = ffi.gc(C.SVM_load(filename), Classes.Algorithm_dtor)
end

Usage:

local svm = cv.ml.SVM()
svm:load('./SVM_MODEL.xml')

[shrubb]

@lymhust
Thank you very much for the effort, and sorry for late reply.
This is a problem indeed, it turns out that we should wrap these :load() functions for each cv::Algorithm subclass.
I'll now do this for cv::ml based on your patch, and later for other packages. Then it will be possible to call svm:load{'svm_model.xml'}.