Search in sources :

Example 1 with SVM

use of org.bytedeco.opencv.opencv_ml.SVM in project qupath by qupath.

the class SVMClassifier method createClassifier.

@Override
protected SVM createClassifier() {
    SVM svm = SVM.create();
    ParameterList params = getParameterList();
    if (params != null) {
        String kernel = (String) params.getChoiceParameterValue("kernel");
        kernel = kernel.toLowerCase();
        if (kernel.equals("linear"))
            svm.setKernel(SVM.LINEAR);
        else if (kernel.equals("polynomial"))
            svm.setKernel(SVM.POLY);
        else if (kernel.equals("rbf"))
            svm.setKernel(SVM.RBF);
        else if (kernel.equals("sigmoid"))
            svm.setKernel(SVM.SIGMOID);
        else if (kernel.equals("chi2"))
            svm.setKernel(SVM.CHI2);
        else if (kernel.equals("histogram intersection"))
            svm.setKernel(SVM.INTER);
        double c = params.getDoubleParameterValue("c");
        if (c > 0)
            svm.setC(c);
        svm.setGamma(params.getDoubleParameterValue("gamma"));
        svm.setDegree(params.getIntParameterValue("degree"));
    }
    return svm;
}
Also used : ParameterList(qupath.lib.plugins.parameters.ParameterList) SVM(org.bytedeco.opencv.opencv_ml.SVM)

Aggregations

SVM (org.bytedeco.opencv.opencv_ml.SVM)1 ParameterList (qupath.lib.plugins.parameters.ParameterList)1