修复OCR BUG

wilinz · Aug 15, 2022 · 58f763f · 58f763f
1 parent cfb5903
commit 58f763f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/autojs/src/main/java/com/stardust/autojs/runtime/api/Paddle.java b/autojs/src/main/java/com/stardust/autojs/runtime/api/Paddle.java
@@ -20,7 +20,7 @@ public void initOcr(Context context, int cpuThreadNum, boolean useSlim) {
     }
 
     public boolean initOcr(Context context, String myModelPath) {
-        return mPredictor.init(context, myModelPath,myModelPath);
+        return mPredictor.init(context, myModelPath);
     }
 
     public List<OcrResult> ocr(ImageWrapper image, int cpuThreadNum, boolean useSlim) {
@@ -48,7 +48,7 @@ public List<OcrResult> ocr(ImageWrapper image, int cpuThreadNum, String myModelP
         if (!mPredictor.isLoaded()) {
             initOcr(GlobalAppContext.get(), myModelPath);
         }
-        return mPredictor.runOcr(bitmap, 4, true);
+        return mPredictor.runOcr(bitmap, cpuThreadNum, true);
     }
 
     public List<OcrResult> ocr(ImageWrapper image, int cpuThreadNum) {
@@ -84,7 +84,7 @@ public void release() {
     private String[] sortResult(List<OcrResult> src) {
         String[] outputResult = new String[src.size()];
         for (int i = 0; i < src.size(); i++) {
-            outputResult[i] = src.get(i).words;
+            outputResult[i] = src.get(i).getWords();
             Log.i("outputResult", outputResult[i].toString()); // show LOG in Logcat panel
         }
         return outputResult;