createConcentricSpheresTestSet.cpp

Go to the documentation of this file.

#include "mexopencv.hpp"
#include "opencv2/ml.hpp"
using namespace std;
using namespace cv;
using namespace cv::ml;

//TODO: bug in 3.1.0, fixed in master
// https://github.com/Itseez/opencv/pull/5965

void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[])
{
    // Check the number of arguments
    nargchk(nrhs==3 && nlhs<=2);

    // Argument vector
    vector<MxArray> rhs(prhs, prhs+nrhs);

    // Process
    int num_samples = rhs[0].toInt(),
        num_features = rhs[1].toInt(),
        num_classes = rhs[2].toInt();
    Mat samples, responses;
    createConcentricSpheresTestSet(num_samples, num_features, num_classes,
        samples, responses);
    plhs[0] = MxArray(samples);
    if (nlhs > 1)
        plhs[1] = MxArray(responses);
}