718c41634f
1.项目后端整体迁移至PaddleOCR-NCNN算法，已通过基本的兼容性测试 2.工程改为使用CMake组织，后续为了更好地兼容第三方库，不再提供QMake工程 3.重整权利声明文件，重整代码工程，确保最小化侵权风险 Log: 切换后端至PaddleOCR-NCNN，切换工程为CMake Change-Id: I4d5d2c5d37505a4a24b389b1a4c5d12f17bfa38c
43 lines
1.2 KiB
Java
43 lines
1.2 KiB
Java
package org.opencv.test.ml;
|
|
|
|
import org.opencv.ml.Ml;
|
|
import org.opencv.ml.SVM;
|
|
import org.opencv.core.Mat;
|
|
import org.opencv.core.MatOfFloat;
|
|
import org.opencv.core.MatOfInt;
|
|
import org.opencv.core.CvType;
|
|
import org.opencv.test.OpenCVTestCase;
|
|
import org.opencv.test.OpenCVTestRunner;
|
|
|
|
public class MLTest extends OpenCVTestCase {
|
|
|
|
public void testSaveLoad() {
|
|
Mat samples = new MatOfFloat(new float[] {
|
|
5.1f, 3.5f, 1.4f, 0.2f,
|
|
4.9f, 3.0f, 1.4f, 0.2f,
|
|
4.7f, 3.2f, 1.3f, 0.2f,
|
|
4.6f, 3.1f, 1.5f, 0.2f,
|
|
5.0f, 3.6f, 1.4f, 0.2f,
|
|
7.0f, 3.2f, 4.7f, 1.4f,
|
|
6.4f, 3.2f, 4.5f, 1.5f,
|
|
6.9f, 3.1f, 4.9f, 1.5f,
|
|
5.5f, 2.3f, 4.0f, 1.3f,
|
|
6.5f, 2.8f, 4.6f, 1.5f
|
|
}).reshape(1, 10);
|
|
Mat responses = new MatOfInt(new int[] {
|
|
0, 0, 0, 0, 0, 1, 1, 1, 1, 1
|
|
}).reshape(1, 10);
|
|
SVM saved = SVM.create();
|
|
assertFalse(saved.isTrained());
|
|
|
|
saved.train(samples, Ml.ROW_SAMPLE, responses);
|
|
assertTrue(saved.isTrained());
|
|
|
|
String filename = OpenCVTestRunner.getTempFileName("yml");
|
|
saved.save(filename);
|
|
SVM loaded = SVM.load(filename);
|
|
assertTrue(loaded.isTrained());
|
|
}
|
|
|
|
}
|