OpenCV通过SVM训练车牌判别（JAVA版）

这是Java版本的OpenCV训练代码，测试集和训练集提供下载连接。这里只贴完整代码。带有注释，理解起来不困难。博主测试后识别率高。数据集下载地址https://download.csdn.net/download/wskfly/12030833

package com.it.opencv;

import java.io.File;
import java.util.ArrayList;

import org.junit.Test;
import org.opencv.core.Core;
import org.opencv.core.CvType;
import org.opencv.core.Mat;
import org.opencv.core.TermCriteria;
import org.opencv.imgcodecs.Imgcodecs;
import org.opencv.ml.Ml;
import org.opencv.ml.SVM;
import org.opencv.ml.TrainData;

public class trainsvm {
	static {
		System.loadLibrary(Core.NATIVE_LIBRARY_NAME);
	}
	//存储训练集
	static ArrayList<Mat> trainingImages = new ArrayList<Mat>();
	//存储标签
	static ArrayList<Integer> trainingLabels = new ArrayList<Integer>();

	public static void main(String[] args) {

		openFile(1, "C:\\Users\\98432\\Desktop\\HasPlate");
		openFile(0, "C:\\Users\\98432\\Desktop\\NoPlate");
		Mat srcImgs = new Mat();
		Mat flags = new Mat(trainingLabels.size(), 1, CvType.CV_32SC1);
		Core.vconcat(trainingImages, srcImgs);
		for (int i = 0; i < trainingLabels.size(); i++) {
			int[] val = { trainingLabels.get(i) };
			flags.put(i, 0, val);
		}
		SVM svm = SVM.create();
		svm.setKernel(SVM.LINEAR);
		svm.setType(SVM.C_SVC);
		svm.setGamma(1);
		svm.setC(1);
		svm.setCoef0(0);
		svm.setNu(0);
		svm.setP(0);
		svm.setTermCriteria(new TermCriteria(TermCriteria.MAX_ITER, 20000, 0.0001));
		TrainData trainData = TrainData.create(srcImgs, Ml.ROW_SAMPLE, flags);
		boolean success = svm.train(trainData);
		System.out.println(success);
		svm.save("E:\\svmTrain\\bp.xml");

	}

	//读取本地文件
	public static void openFile(int flag, String path) {
		File file = new File(path);
		File[] files = file.listFiles();
		for (File file2 : files) {
			Mat input = Imgcodecs.imread(file2.getPath(), 0);
			input.convertTo(input, CvType.CV_32FC1);
			Mat reshape = input.reshape(0, 1);
			// System.out.println(input.dump());
			trainingImages.add(reshape);
			trainingLabels.add(flag);
		}
	}

	@Test
	public void test1() {

		// 测试训练的效果
		SVM svm = SVM.load("E:\\svmTrain\\bp.xml");
        Mat responseMat = new Mat(); 
        Mat imread = Imgcodecs.imread("C:\\Users\\98432\\Desktop\\platetest\\NoPlate\\S22_KG2187_3.jpg", 0);
        imread.convertTo(imread, CvType.CV_32FC1);
        Mat reshape = imread.reshape(0, 1);
        svm.predict(reshape, responseMat, 0);  
		System.out.println(responseMat.dump());
        
        
	}

}