PPOCR-ONNX使用说明

1
function test_http_ppocr() {
2
    if (!startEnv()) {
3
      loge("自动化启动失败,结束脚本")
4
      exit()
5
    }
6
    if (!image.requestScreenCapture(10000, 0)) {
7
      loge("申请截图权限失败,检查是否开启后台弹出,悬浮框等权限")
8
      exit()
9
    }
10
    //申请完权限至少等1s(垃圾设备多加点)再截图,否则会截不到图
11
    sleep(1000)
12

13
    for (let i = 0; i < 1; i++) {
14
      let img = image.captureFullScreenEx()
15
      if (!img) {
16
        loge("截图失败")
17
        sleep(1000)
18
        continue
19
      }
20
      // 使用的识别模型 paddleOcrOnnxV5 代表PPOCR-V5 ， paddleOcrOnnxV4 代表 PPOCR-V4
21
      let ocrType = "paddleOcrOnnxV5";
22
      // padding 图像外接白框，用于提升识别率，文字框没有正确框住所有文字时，增加此值。默认 10。
23
      let padding = 50;
24
      // maxSideLen 如果输入图像的最大边大于max_side_len，则会按宽高比，将最大边缩放到max_side_len。默认为 0
25
      let maxSideLen = 0;
26
      try {
27

28
        let url = `http://192.168.2.19:9022/devapi/ocrOnnxFile?ocrType=${ocrType}&padding=${padding}&maxSideLen=${maxSideLen}`
29
        console.log(`request url ${url}`)
30
        // 保存文件到 /sdcard/tb.png
31
        image.saveTo(img, "/sdcard/tb.png")
32
        let file = {"file": "/sdcard/tb.png"}
33
        console.time(1)
34
        let result = http.httpPost(url, {}, file, 20 * 1000, {})
35
        logd("请求耗时: {} ms", console.timeEnd(1))
36
        if (!result) {
37
          logw("未识别到结果")
38
          sleep(1000)
39
          continue
40
        }
41
        logd("ocr结果-> " + result)
42
        result = JSON.parse(result).data;
43
        for (let i = 0; i < result.length; i++) {
44
          let value = result[i]
45
          logd("文字 : " + value.label+ " confidence: "+value.confidence + " x: " + value.x + " y: " + value.y + " width: " + value.width + " height: " + value.height)
46
        }
47
        sleep(1000)
48
      } catch (e) {
49
        logd(e)
50
      } finally {
51
        image.recycle(img)
52
      }
53
    }
54
  }
55

56
  test_http_ppocr()

1
function test_http_ppocr_base() {
2

3
    if (!startEnv()) {
4
        loge("自动化启动失败,结束脚本")
5
        exit()
6
    }
7
    if (!image.requestScreenCapture(10000, 0)) {
8
        loge("申请截图权限失败,检查是否开启后台弹出,悬浮框等权限")
9
        exit()
10
    }
11
    //申请完权限至少等1s(垃圾设备多加点)再截图,否则会截不到图
12
    sleep(1000)
13

14
    for (let i = 0; i < 1; i++) {
15
        let img = image.captureFullScreenEx()
16
        if (!img) {
17
            loge("截图失败")
18
            sleep(1000)
19
            continue
20
        }
21
        // 使用的识别模型 paddleOcrOnnxV5 代表PPOCR-V5 ， paddleOcrOnnxV4 代表 PPOCR-V4
22
        let ocrType = "paddleOcrOnnxV5";
23
        // padding 图像外接白框，用于提升识别率，文字框没有正确框住所有文字时，增加此值。默认 10。
24
        let padding = 50;
25
        // maxSideLen 如果输入图像的最大边大于max_side_len，则会按宽高比，将最大边缩放到max_side_len。默认为 0
26
        let maxSideLen = 0;
27
        try {
28

29
            let url = `http://192.168.2.19:9022/devapi/ocrOnnxBase64`
30
            console.log(`request url ${url}`)
31
            let base = image.toBase64(img)
32
            let data = {
33
                "ocrType":ocrType,
34
                "padding":padding,
35
                "maxSideLen":maxSideLen,
36
                "data":base,
37
            }
38
            console.time(1)
39
            let result = http.postJSON(url, data, 20 * 1000, {})
40
            logd("请求耗时: {} ms", console.timeEnd(1))
41
            if (!result) {
42
                logw("未识别到结果")
43
                sleep(1000)
44
                continue
45
            }
46
            logd("ocr结果-> " + result)
47
            result = JSON.parse(result).data;
48
            for (let i = 0; i < result.length; i++) {
49
                let value = result[i]
50
                logd("文字 : " + value.label+ " confidence: "+value.confidence + " x: " + value.x + " y: " + value.y + " width: " + value.width + " height: " + value.height)
51
            }
52
            sleep(1000)
53
        } catch (e) {
54
            logd(e)
55
        } finally {
56
            image.recycle(img)
57
        }
58
    }
59
}
60
test_http_ppocr_base();