show error info when load model error

Author: litongjava

readme.md

@@ -11,6 +11,9 @@
 
 ## 开源地址
 [gitee](https://gitee.com/ppnt/tools-ocr) | [github](https://github.com/litongjava/tools-ocr)
+
+## required
+- Mac OS X 12.6 因为依赖djl 0.25.0
 ## 安装
 > - **安装路径请勿包含中文字符**；
 > - 本程序使用 JavaFX 开发，提供的安装包中已经包含了Java
@@ -43,10 +46,10 @@ wget https://github.com/litongjava/tools-ocr/releases/download/model-ppocr-v4/ch
 ```
 解压模型
 ```
-mkdir models/ch_PP-OCRv4_rec_infer-onnx
-mkdir models/ch_PP-OCRv4_det_infer-onnx
-unzip /Users/mac/Downloads/ch_PP-OCRv4_rec_infer-onnx.zip -d models/ch_PP-OCRv4_rec_infer-onnx
-unzip /Users/mac/Downloads/ch_PP-OCRv4_det_infer-onnx.zip -d models/ch_PP-OCRv4_det_infer-onnx
+mkdir models/ch_PP-OCRv4_rec_infer
+mkdir models/ch_PP-OCRv4_det_infer
+unzip /Users/mac/Downloads/ch_PP-OCRv4_rec_infer-onnx.zip -d models/ch_PP-OCRv4_rec_infer
+unzip /Users/mac/Downloads/ch_PP-OCRv4_det_infer-onnx.zip -d models/ch_PP-OCRv4_det_infer
 ```
 
 ### 构建程序

src/main/java/com/luooqi/ocr/OcrApp.java

@@ -19,7 +19,7 @@ public static void main(String[] args) {
   @Override
   public void init() throws Exception {
     super.init();
-    InitConfig.init();
+    //InitConfig.init();
   }
 
 

src/main/java/com/luooqi/ocr/config/InitConfig.java

@@ -1,6 +1,5 @@
 package com.luooqi.ocr.config;
 
-import com.luooqi.ocr.local.PaddlePaddleOCRV4;
 import com.luooqi.ocr.utils.GlobalKeyListener;
 import com.luooqi.ocr.utils.VoidDispatchService;
 import org.jnativehook.GlobalScreen;
@@ -23,7 +22,7 @@ public static void init() {
 //    map.put(ConfigKeys.recName, "ch_PP-OCRv3_rec_infer");
 //    map.put(ConfigKeys.keysName, "ppocr_keys_v1.txt");
 //    projectConfig.batchPut(map);
-    PaddlePaddleOCRV4.INSTANCE.init();
+
 
   }
 

src/main/java/com/luooqi/ocr/local/PaddlePaddleOCRV4.java

@@ -26,36 +26,30 @@
  */
 public enum PaddlePaddleOCRV4 {
   INSTANCE;
-  private OcrV4Detection detection;
-  private OcrV4Recognition recognition;
-  private Predictor<Image, NDList> detector;
-  private Predictor<Image, String> recognizer;
-  private NDManager manager;
+  private static OcrV4Detection detection;
+  private static OcrV4Recognition recognition;
+  private static Predictor<Image, NDList> detector;
+  private static Predictor<Image, String> recognizer;
+  private static NDManager manager;
 
   PaddlePaddleOCRV4() {
+  }
+
+
+  //noting not to do.but init
+  public static void init() throws ModelNotFoundException, MalformedModelException, IOException {
     detection = new OcrV4Detection();
     recognition = new OcrV4Recognition();
     ZooModel detectionModel = null;
     ZooModel recognitionModel = null;
-    try {
-      detectionModel = ModelZoo.loadModel(detection.chDetCriteria());
-      recognitionModel = ModelZoo.loadModel(recognition.chRecCriteria());
-    } catch (IOException e) {
-      e.printStackTrace();
-    } catch (ModelNotFoundException e) {
-      e.printStackTrace();
-    } catch (MalformedModelException e) {
-      e.printStackTrace();
-    }
+
+    detectionModel = ModelZoo.loadModel(detection.chDetCriteria());
+    recognitionModel = ModelZoo.loadModel(recognition.chRecCriteria());
+
     detector = detectionModel.newPredictor();
 
     recognizer = recognitionModel.newPredictor();
     manager = NDManager.newBaseManager();
-  }
-
-
-  //noting not to do.but init
-  public void init() {
 
   }
 

src/main/java/com/luooqi/ocr/windows/MainForm.java

@@ -1,11 +1,14 @@
 package com.luooqi.ocr.windows;
 
+import ai.djl.MalformedModelException;
+import ai.djl.repository.zoo.ModelNotFoundException;
 import cn.hutool.core.io.FileTypeUtil;
 import cn.hutool.core.thread.ThreadUtil;
 import cn.hutool.core.util.StrUtil;
 import cn.hutool.log.StaticLog;
 import com.luooqi.ocr.config.InitConfig;
 import com.luooqi.ocr.controller.ProcessController;
+import com.luooqi.ocr.local.PaddlePaddleOCRV4;
 import com.luooqi.ocr.model.CaptureInfo;
 import com.luooqi.ocr.model.StageInfo;
 import com.luooqi.ocr.snap.ScreenCapture;
@@ -30,6 +33,7 @@
 
 import java.awt.image.BufferedImage;
 import java.io.File;
+import java.io.IOException;
 import java.util.HashMap;
 import java.util.Map;
 
@@ -86,6 +90,16 @@ public void init(Stage primaryStage) {
     CommUtils.initStage(primaryStage);
     mainScene = new Scene(root, 670, 470);
     stage.setScene(mainScene);
+    //启动引擎,加载模型,如果模型加载错误下屏幕显示错误
+    try {
+      PaddlePaddleOCRV4.init();
+    } catch (ModelNotFoundException e) {
+      textArea.setText("加载模型出现错误" + e.getMessage());
+    } catch (MalformedModelException e) {
+      textArea.setText("加载模型出现错误" + e.getMessage());
+    } catch (IOException e) {
+      textArea.setText("加载模型出现错误" + e.getMessage());
+    }
   }
 
   private TextArea getCenter() {