Merge pull request #76 from litongjava/rapidocr-onnx

Rapidocr onnx

Author: litongjava

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

@@ -6,6 +6,7 @@
 
 import io.github.mymonstercat.Model;
 import io.github.mymonstercat.ocr.InferenceEngine;
+import io.github.mymonstercat.ocr.config.HardwareConfig;
 
 /**
  * Created by [email protected] on 11/23/2023_2:09 AM
@@ -21,14 +22,15 @@ public enum PaddlePaddleOCRV4 {
 
   // noting not to do.but init
   public static void init() {
-    engine = InferenceEngine.getInstance(Model.ONNX_PPOCR_V4_SERVER);
+    HardwareConfig onnxConfig = HardwareConfig.getOnnxConfig();
+    onnxConfig.setNumThread(2);
+    engine = InferenceEngine.getInstance(Model.ONNX_PPOCR_V4_SERVER, onnxConfig);
   }
 
   public OcrResult ocr(File imageFile) {
     return engine.runOcr(imageFile.getAbsolutePath());
   }
 
   public void close() {
-
   }
 }

src/test/java/com/litongjava/RapidOcrTest.java

@@ -4,11 +4,19 @@
 
 import io.github.mymonstercat.Model;
 import io.github.mymonstercat.ocr.InferenceEngine;
+import io.github.mymonstercat.ocr.config.HardwareConfig;
 
 public class RapidOcrTest {
   public static void main(String[] args) {
-    InferenceEngine engine = InferenceEngine.getInstance(Model.ONNX_PPOCR_V4_SERVER);
-    OcrResult ocrResult = engine.runOcr("images/01.png");
+    String imagePath = "C:\\Users\\Administrator\\Desktop\\01.jpg";
+
+    // init
+    HardwareConfig onnxConfig = HardwareConfig.getOnnxConfig();
+    onnxConfig.setNumThread(2);
+    InferenceEngine engine = InferenceEngine.getInstance(Model.ONNX_PPOCR_V4_SERVER, onnxConfig);
+
+    // run
+    OcrResult ocrResult = engine.runOcr(imagePath);    
     System.out.println(ocrResult.getStrRes().trim());
   }
 }