Wie trainiere ich eine SVM mit opencv basierend auf einer Reihe von Bildern?

Question

Ich habe einen Ordner von Positiven und eine andere von Negativen Bilder im JPG-Format, und ich möchte ein SVM basierend trainieren auf, dass die Bilder, ich habe nach dem getan, aber ich erhalte eine Fehlermeldung:

Mat classes = new Mat(); 
Mat trainingData = new Mat(); 

Mat trainingImages = new Mat(); 
Mat trainingLabels = new Mat(); 

CvSVM clasificador; 

for (File file : new File(path + "positives/").listFiles()) { 
     Mat img = Highgui.imread(file.getAbsolutePath()); 
     img.reshape(1, 1); 

     trainingImages.push_back(img); 
     trainingLabels.push_back(Mat.ones(new Size(1, 1), CvType.CV_32FC1)); 
    } 

    for (File file : new File(path + "negatives/").listFiles()) { 
     Mat img = Highgui.imread(file.getAbsolutePath()); 
     img.reshape(1, 1); 

     trainingImages.push_back(img); 
     trainingLabels.push_back(Mat.zeros(new Size(1, 1), CvType.CV_32FC1)); 
    } 

    trainingImages.copyTo(trainingData); 
    trainingData.convertTo(trainingData, CvType.CV_32FC1); 
    trainingLabels.copyTo(classes); 

    CvSVMParams params = new CvSVMParams(); 
    params.set_kernel_type(CvSVM.LINEAR); 

    clasificador = new CvSVM(trainingData, classes, new Mat(), new Mat(), params); 

Als ich versuche zu laufen das erhalte ich:

OpenCV Error: Bad argument (train data must be floating-point matrix) in cvCheckTrainData, file ..\..\..\src\opencv\modules\ml\src\inner_functions.cpp, line 857 
Exception in thread "main" CvException [org.opencv.core.CvException: ..\..\..\src\opencv\modules\ml\src\inner_functions.cpp:857: error: (-5) train data must be floating-point matrix in function cvCheckTrainData 
] 
    at org.opencv.ml.CvSVM.CvSVM_1(Native Method) 
    at org.opencv.ml.CvSVM.<init>(CvSVM.java:80) 

Ich kann es nicht schaffen, die SVM zu trainieren, keine Ahnung? Danke

Answer 1

Unter der Annahme, dass Sie wissen, was Sie durch Umformen ein Bild machen und die Verwendung dieser SVM zu trainieren, die wahrscheinlichste Ursache dafür ist, dass Ihr

Mat img = Highgui.imread(file.getAbsolutePath()); 

nicht tatsächlich um ein Bild zu lesen, um eine Matrix zu erzeugen img mit null data Eigenschaft, die in dem Code OpenCV schließlich auslösen folgend wird:

// check parameter types and sizes 
if(!CV_IS_MAT(train_data) || CV_MAT_TYPE(train_data->type) != CV_32FC1) 
    CV_ERROR(CV_StsBadArg, "train data must be floating-point matrix"); 

Grundsätzlich train_data nicht die erste Bedingung (eine gültige Matrix ist), anstatt sie nicht die zweite Bedingung (vom Typ CV_32FC1).

Darüber hinaus funktioniert, obwohl Umformen auf dem *this Objekt funktioniert, es wie ein Filter und seine Wirkung ist nicht dauerhaft. Wenn es in einer einzelnen Anweisung verwendet wird, ohne sofort verwendet oder einer anderen Variablen zugewiesen zu werden, ist es nutzlos. Ändern Sie die folgenden Zeilen im Code:

img.reshape(1, 1); 
trainingImages.push_back(img); 

zu:

trainingImages.push_back(img.reshape(1, 1)); 

Answer 2

Genau wie der Fehler sagt, müssen Sie den Typ Ihrer Matrix ändern, vom Integer-Typ, wahrscheinlich CV_8U, zum Gleitkomma, CV_32F oder CV_64F. Um es zu tun Sie können cv::Mat::convertTo() verwenden. Here ist ein bisschen über Tiefen und Arten von Matrizen.