From 38de48be7d2f1609b278484d151202d3ef122c05 Mon Sep 17 00:00:00 2001 From: litongjava Date: Sun, 28 Apr 2024 06:20:04 -1000 Subject: [PATCH 1/2] set default thread is 2 --- src/main/java/com/luooqi/ocr/local/PaddlePaddleOCRV4.java | 5 ++++- src/test/java/com/litongjava/RapidOcrTest.java | 7 +++++-- 2 files changed, 9 insertions(+), 3 deletions(-) diff --git a/src/main/java/com/luooqi/ocr/local/PaddlePaddleOCRV4.java b/src/main/java/com/luooqi/ocr/local/PaddlePaddleOCRV4.java index ff77b49..758baa0 100644 --- a/src/main/java/com/luooqi/ocr/local/PaddlePaddleOCRV4.java +++ b/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 litonglinux@qq.com on 11/23/2023_2:09 AM @@ -21,7 +22,9 @@ 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) { diff --git a/src/test/java/com/litongjava/RapidOcrTest.java b/src/test/java/com/litongjava/RapidOcrTest.java index 27090aa..4349ac8 100644 --- a/src/test/java/com/litongjava/RapidOcrTest.java +++ b/src/test/java/com/litongjava/RapidOcrTest.java @@ -4,11 +4,14 @@ 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"); + HardwareConfig onnxConfig = HardwareConfig.getOnnxConfig(); + onnxConfig.setNumThread(2); + InferenceEngine engine = InferenceEngine.getInstance(Model.ONNX_PPOCR_V4_SERVER, onnxConfig); + OcrResult ocrResult = engine.runOcr("C:\\Users\\Administrator\\Desktop\\01.jpg"); System.out.println(ocrResult.getStrRes().trim()); } } From e40f8d7f45df3be3fba4b85231562220db2431be Mon Sep 17 00:00:00 2001 From: litongjava Date: Sun, 28 Apr 2024 07:06:12 -1000 Subject: [PATCH 2/2] change thread --- src/main/java/com/luooqi/ocr/local/PaddlePaddleOCRV4.java | 1 - src/test/java/com/litongjava/RapidOcrTest.java | 7 ++++++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/src/main/java/com/luooqi/ocr/local/PaddlePaddleOCRV4.java b/src/main/java/com/luooqi/ocr/local/PaddlePaddleOCRV4.java index 758baa0..878ddc5 100644 --- a/src/main/java/com/luooqi/ocr/local/PaddlePaddleOCRV4.java +++ b/src/main/java/com/luooqi/ocr/local/PaddlePaddleOCRV4.java @@ -32,6 +32,5 @@ public OcrResult ocr(File imageFile) { } public void close() { - } } diff --git a/src/test/java/com/litongjava/RapidOcrTest.java b/src/test/java/com/litongjava/RapidOcrTest.java index 4349ac8..0bfba20 100644 --- a/src/test/java/com/litongjava/RapidOcrTest.java +++ b/src/test/java/com/litongjava/RapidOcrTest.java @@ -8,10 +8,15 @@ public class RapidOcrTest { public static void main(String[] args) { + 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); - OcrResult ocrResult = engine.runOcr("C:\\Users\\Administrator\\Desktop\\01.jpg"); + + // run + OcrResult ocrResult = engine.runOcr(imagePath); System.out.println(ocrResult.getStrRes().trim()); } }