forked from TonyJiangWJ/Ant-Forest
-
Notifications
You must be signed in to change notification settings - Fork 0
/
LocalOcrUtil.js
27 lines (25 loc) · 1.04 KB
/
LocalOcrUtil.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
let { config } = require('../config.js')(runtime, global)
let singletonRequire = require('../lib/SingletonRequirer.js')(runtime, global)
let logUtils = singletonRequire('LogUtils')
let paddleOcrUtil = singletonRequire('PaddleOcrUtil')
let mlkitOcrUtil = singletonRequire('MlkitOcrUtil')
let localOcr = null
logUtils.debugInfo(['当前本地OCR优先级为:{}', config.local_ocr_priority])
if (config.local_ocr_priority == 'mlkit' || config.local_ocr_priority == 'auto') {
localOcr = mlkitOcrUtil.enabled ? mlkitOcrUtil : paddleOcrUtil.enabled ? paddleOcrUtil : null
} else if (config.local_ocr_priority == 'paddle') {
if (!paddleOcrUtil.initialized) {
paddleOcrUtil.init()
}
localOcr = paddleOcrUtil.enabled ? paddleOcrUtil : mlkitOcrUtil.enabled ? mlkitOcrUtil : null
}
if (localOcr == null) {
localOcr = {
enabled: false,
type: '不受支持',
recognize: () => '',
recognizeWithBounds: () => []
}
}
logUtils.debugInfo(['当前启用的OCR为:{} 是否支持:{}', localOcr.type, localOcr.enabled])
module.exports = localOcr