Created
July 6, 2016 10:54
-
-
Save clive819/07e6ee955271c39fba9ac9eab43da297 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
/*public static void main(String args[]){ | |
SVM svm = SVM.create(); | |
HOGDescriptor hog = new HOGDescriptor(new Size(64, 128), new Size(16, 16), new Size(8, 8), new Size(8, 8), 9); | |
Mat vector = new Mat(); | |
Mat label = new Mat(); | |
int dim = 0; | |
for (int i=0; i<100; i++){ | |
Mat img = Imgcodecs.imread("/Users/Clive/Desktop/trainHOG-master/pos/"+i+".png"); | |
MatOfFloat factor = new MatOfFloat(); | |
MatOfPoint locations =new MatOfPoint(); | |
hog.compute(img, factor, new Size(64, 128), new Size(8, 8), locations); | |
if (i == 0){ | |
dim = factor.toArray().length; | |
vector = Mat.zeros(200, dim, CvType.CV_32FC1); | |
label = Mat.zeros(200, 1, CvType.CV_32SC1); | |
} | |
for (int j=0; j<dim; j++){ | |
vector.put(i, j, factor.toArray()[j]); | |
} | |
label.put(i, 0, 1); | |
} | |
for (int i=0; i<100; i++){ | |
Mat img = Imgcodecs.imread("/Users/Clive/Desktop/trainHOG-master/neg/"+i+".png"); | |
MatOfFloat factor = new MatOfFloat(); | |
MatOfPoint locations =new MatOfPoint(); | |
hog.compute(img, factor, new Size(64, 128), new Size(8, 8), locations); | |
for (int j=0; j<dim; j++){ | |
vector.put(i+100, j, factor.toArray()[j]); | |
} | |
label.put(i+100, 0, -1); | |
} | |
svm.train(vector, Ml.ROW_SAMPLE, label); | |
svm.save("data.xml"); | |
System.out.println("Done!"); | |
}*/ |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment