diff --git a/src/main/java/com/luooqi/ocr/local/PaddlePaddleOCRV4.java b/src/main/java/com/luooqi/ocr/local/PaddlePaddleOCRV4.java index ff77b49..878ddc5 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) { @@ -29,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 27090aa..0bfba20 100644 --- a/src/test/java/com/litongjava/RapidOcrTest.java +++ b/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()); } }