no commit message

src/main/java/com/yuxue/train/ANNTrain.java

@@ -121,12 +121,10 @@ public class ANNTrain {
         for (int i = 0; i < Constant.numCharacter; i++) {
             String str = DEFAULT_PATH + "learn/" + Constant.strCharacters[i];
             Vector<String> files = new Vector<String>();
-            FileUtil.getFiles(str, files);
+            FileUtil.getFiles(str, files);  // 文件名不能包含中文
             
-            int size = (int) files.size();
-            for (int j = 0; j < size; j++) {
-                Mat img = Imgcodecs.imread(files.get(j), 0);
-                // System.err.println(files.get(j)); // 文件名不能包含中文
+            for (String filePath : files) {
+                Mat img = Imgcodecs.imread(filePath);
                 Mat f = features(img, _predictsize);
                 samples.push_back(f);
                 trainingLabels.add(i); // 每一幅字符图片所对应的字符类别索引下标
@@ -138,18 +136,14 @@ public class ANNTrain {
             String str = DEFAULT_PATH + "learn/" + Constant.strChinese[i];
             Vector<String> files = new Vector<String>();
             FileUtil.getFiles(str, files);
-
-            int size = (int) files.size();
-            for (int j = 0; j < size; j++) {
-                Mat img = Imgcodecs.imread(files.get(j), 0);
-                // System.err.println(files.get(j));   // 文件名不能包含中文
+            for (String filePath : files) {
+                Mat img = Imgcodecs.imread(filePath);
                 Mat f = features(img, _predictsize);
                 samples.push_back(f);
-                trainingLabels.add(i + Constant.numCharacter);
+                trainingLabels.add(i + Constant.numCharacter); // 每一幅字符图片所对应的字符类别索引下标
             }
         }
 
-
         //440   vhist.length + hhist.length + lowData.cols() * lowData.rows();
         // CV_32FC1 CV_32SC1 CV_32F
         Mat classes = new Mat(trainingLabels.size(), Constant.numAll, CvType.CV_32F);
@@ -168,7 +162,7 @@ public class ANNTrain {
         layers.put(0, 1, _neurons);
         layers.put(0, 2, classes.cols());
 
-        System.out.println(layers);
+        // System.out.println(layers);
 
         ann.setLayerSizes(layers);
         ann.setActivationFunction(ANN_MLP.SIGMOID_SYM, 1, 1);
@@ -191,6 +185,7 @@ public class ANNTrain {
         Vector<String> files = new Vector<String>();
         FileUtil.getFiles(DEFAULT_PATH + "test/", files);
 
+        String plate = "";
         for (String string : files) {
             Mat img = Imgcodecs.imread(string, 0);
             Mat f = features(img, Constant.predictSize);
@@ -199,12 +194,21 @@ public class ANNTrain {
             // 440 predictSize = 20;
             Mat output = new Mat(1, 140, CvType.CV_32F);
             //ann.predict(f, output, 0);  // 预测结果
-            System.err.println(string + "===>" + (int) ann.predict(f, output, 0));
 
             // ann.predict(f, output, 0);
             // System.err.println(string + "===>" + output.get(0, 0)[0]);
 
+            int index = (int) ann.predict(f, output, 0);
+            System.err.println(string + "===>" + index);
+
+            if (index < Constant.numCharacter) {
+                plate += String.valueOf(Constant.strCharacters[index]);
+            } else {
+                String s = Constant.strChinese[index - Constant.numCharacter];
+                plate += Constant.KEY_CHINESE_MAP.get(s);
+            }
         }
+        System.err.println("===>" + plate);
     }
 
     public static void main(String[] args) {

src/main/java/com/yuxue/train/ANNTrain1.java

@@ -128,7 +128,7 @@ public class ANNTrain1 {
             int index = (int) ann.predict(f, output, 0);
             
             String result = "";
-            if (index <= Constant.numCharacter) {
+            if (index < Constant.numCharacter) {
                 result = String.valueOf(Constant.strCharacters[index]);
             } else {
                 String s = Constant.strChinese[index - Constant.numCharacter];

src/main/java/com/yuxue/util/ImageUtil.java

@@ -71,7 +71,7 @@ public class ImageUtil {
         Instant start = Instant.now();
         String tempPath = DEFAULT_BASE_TEST_PATH + "test/";
          String filename = tempPath + "/100_yuantu.jpg";
-        filename = tempPath + "/100_yuantu2.jpg";
+        filename = tempPath + "/100_yuantu1.jpg";
         // filename = tempPath + "/109_crop_0.png";
 
         Mat src = Imgcodecs.imread(filename);

src/main/java/com/yuxue/util/PlateUtil.java

@@ -24,6 +24,7 @@ import com.google.common.collect.Lists;
 import com.google.common.collect.Maps;
 import com.yuxue.constant.Constant;
 import com.yuxue.enumtype.PlateColor;
+import com.yuxue.train.ANNTrain;
 import com.yuxue.train.SVMTrain;
 
 
@@ -59,6 +60,7 @@ public class PlateUtil {
             index ++;
         }
         
+        // 这个位置加载模型文件会报错，暂时没时间定位啥问题报错
         /*loadSvmModel("D:/PlateDetect/train/plate_detect_svm/svm2.xml");
         loadAnnModel("D:/PlateDetect/train/chars_recognise_ann/ann.xml");*/
     }
@@ -288,15 +290,33 @@ public class PlateUtil {
         Vector<Rect> sorted = new Vector<Rect>();
         sortRect(rt, sorted);
 
+        String plate = "";
         Vector<Mat> dst = new Vector<Mat>();
+        
+        ANNTrain annT = new ANNTrain();
         for (int i = 0; i < sorted.size(); i++) {
             Mat img_crop = new Mat(threshold, sorted.get(i));
             img_crop = preprocessChar(img_crop);
             dst.add(img_crop);
-            Imgcodecs.imwrite(tempPath + debugMap.get("plateCrop") + "_plateCrop_" + i + ".jpg", img_crop);
-        }
+            if(debug) {
+                Imgcodecs.imwrite(tempPath + debugMap.get("plateCrop") + "_plateCrop_" + i + ".jpg", img_crop);
+            }
+            
             
+            Mat f = annT.features(img_crop, Constant.predictSize);
             
+            // 字符预测
+            Mat output = new Mat(1, 140, CvType.CV_32F);
+            int index = (int) ann.predict(f, output, 0);
+            
+            if (index < Constant.numCharacter) {
+                plate += String.valueOf(Constant.strCharacters[index]);
+            } else {
+                String s = Constant.strChinese[index - Constant.numCharacter];
+                plate += Constant.KEY_CHINESE_MAP.get(s);
+            }
+        }
+        System.err.println("===>" + plate);
         
         return;
     }