110 lines
3.1 KiB
Python
110 lines
3.1 KiB
Python
import os
|
||
import psutil
|
||
from plugin_i18n import Translator
|
||
|
||
tr = Translator(__file__, "i18n.csv")
|
||
|
||
# 模块配置路径
|
||
MODELS_CONFIGS = "/models/configs.txt"
|
||
# 保存语言字典
|
||
LangDict = {}
|
||
|
||
|
||
# 动态获取模型库列表
|
||
def _getlanguageList():
|
||
global LangDict
|
||
"""configs.txt 格式示例：
|
||
简体中文(V4)
|
||
ch_PP-OCRv4_det_infer.onnx
|
||
ch_ppocr_mobile_v2.0_cls_infer.onnx
|
||
rec_ch_PP-OCRv4_infer.onnx
|
||
dict_chinese.txt
|
||
|
||
"""
|
||
optionsList = []
|
||
configsPath = os.path.dirname(os.path.abspath(__file__)) + MODELS_CONFIGS
|
||
try:
|
||
with open(configsPath, "r", encoding="utf-8") as file:
|
||
content = file.read()
|
||
parts = content.split("\n\n")
|
||
for part in parts:
|
||
items = part.split("\n")
|
||
if len(items) == 5:
|
||
title, det, cls, rec, keys = items
|
||
LangDict[title] = {
|
||
"det": det,
|
||
"cls": cls,
|
||
"rec": rec,
|
||
"keys": keys,
|
||
}
|
||
optionsList.append([title, title])
|
||
return optionsList
|
||
except FileNotFoundError:
|
||
print("[Error] RapidOCR配置文件configs不存在，请检查文件路径是否正确。", configsPath)
|
||
except IOError:
|
||
print("[Error] RapidOCR配置文件configs无法打开或读取。")
|
||
return []
|
||
|
||
|
||
_LanguageList = _getlanguageList()
|
||
|
||
|
||
# 获取最佳线程数
|
||
def _getThreads():
|
||
try:
|
||
phyCore = psutil.cpu_count(logical=False) # 物理核心数
|
||
lgiCore = psutil.cpu_count(logical=True) # 逻辑核心数
|
||
if (
|
||
not isinstance(phyCore, int)
|
||
or not isinstance(lgiCore, int)
|
||
or lgiCore < phyCore
|
||
):
|
||
raise ValueError("核心数计算异常")
|
||
# 物理核数=逻辑核数，返回逻辑核数
|
||
if phyCore * 2 == lgiCore or phyCore == lgiCore:
|
||
return lgiCore
|
||
# 大小核处理器，返回大核线程数
|
||
big = lgiCore - phyCore
|
||
return big * 2
|
||
except Exception as e:
|
||
print("[Warning] 无法获取CPU核心数！", e)
|
||
return 4
|
||
|
||
|
||
_threads = _getThreads()
|
||
|
||
globalOptions = {
|
||
"title": tr("RapidOCR（本地）"),
|
||
"type": "group",
|
||
"numThread": {
|
||
"title": tr("线程数"),
|
||
"default": _threads,
|
||
"min": 1,
|
||
"isInt": True,
|
||
},
|
||
}
|
||
|
||
localOptions = {
|
||
"title": tr("文字识别（RapidOCR）"),
|
||
"type": "group",
|
||
"language": {
|
||
"title": tr("语言/模型库"),
|
||
"optionsList": _LanguageList,
|
||
},
|
||
"angle": {
|
||
"title": tr("纠正文本方向"),
|
||
"default": False,
|
||
"toolTip": tr("启用方向分类，识别倾斜或倒置的文本。可能降低识别速度。"),
|
||
},
|
||
"maxSideLen": {
|
||
"title": tr("限制图像边长"),
|
||
"optionsList": [
|
||
[1024, "1024 " + tr("（默认）")],
|
||
[2048, "2048"],
|
||
[4096, "4096"],
|
||
[999999, tr("无限制")],
|
||
],
|
||
"toolTip": tr("将边长大于该值的图片进行压缩，可以提高识别速度。可能降低识别精度。"),
|
||
},
|
||
}
|