mexopencv_ml.hpp

Go to the documentation of this file.

#ifndef MEXOPENCV_ML_HPP
#define MEXOPENCV_ML_HPP

#include "mexopencv.hpp"
#include "opencv2/ml.hpp"


// ==================== XXX ====================

MxArray toStruct(const std::vector<cv::ml::DTrees::Node>& nodes);

MxArray toStruct(const std::vector<cv::ml::DTrees::Split>& splits);


// ==================== XXX ====================

cv::Ptr<cv::ml::TrainData> createTrainData(
    const cv::Mat& samples, const cv::Mat& responses,
    std::vector<MxArray>::const_iterator first,
    std::vector<MxArray>::const_iterator last);

cv::Ptr<cv::ml::TrainData> loadTrainData(
    const std::string& filename,
    std::vector<MxArray>::const_iterator first,
    std::vector<MxArray>::const_iterator last);

#endif