sui_feng/AzurLaneAutoScript
1
0
Fork 0
mirror of https://github.com/sui-feng-cb/AzurLaneAutoScript1.git synced 2026-08-12 19:01:47 +08:00
Code Activity

Opt: use ONNX model converted from MXNet for faster OCR

Browse Source
This commit is contained in:
positnuec
2026-08-02 13:17:41 +08:00
parent 6d9534c41d
commit bd7d2e875e
10 changed files with 699 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
module/ocr/models.py
View File

@@ -1,65 +1,76 @@
from module.base.decorator import cached_property
from module.webui.setting import State
def _ocr_class():
if not State.deploy_config.UseOcrOnnx:
from module.ocr.al_ocr import AlOcr
return AlOcr
from module.ocr.al_ocr import AlOcrOnnx
return AlOcrOnnx
class OcrModel:
@cached_property
def azur_lane(self):
# Folder: ./bin/cnocr_models/azur_lane
# Size: 3.25MB
# Size: 3.25MB (MXNet) / 3.34MB (ONNX)
# Model: densenet-lite-gru
# Epoch: 15
# Validation accuracy: 99.43%
# Font: Impact, AgencyFB-Regular, MStiffHeiHK-UltraBold
# Charset: 0123456789ABCDEFGHIJKLMNPQRSTUVWXYZ:/- (Letter 'O' and <space> is not included)
# _num_classes: 39
from module.ocr.al_ocr import AlOcr
return AlOcr(model_name='densenet-lite-gru', model_epoch=15, root='./bin/cnocr_models/azur_lane',
name='azur_lane')
return _ocr_class()(model_name='densenet-lite-gru', model_epoch=15,
root='./bin/cnocr_models/azur_lane', name='azur_lane')
@cached_property
def azur_lane_jp(self):
# Folder: ./bin/cnocr_models/azur_lane_jp
# Size: 3.25MB
# Size: 3.25MB (MXNet) / 3.34MB (ONNX)
# Model: densenet-lite-gru
# Epoch: 20
# Validation accuracy: 99.01%
# Font: Impact, VibeMO Compressed Pro Thin, Folk R, Source Han Serif JP
# Charset: 0123456789ABCDEFGHIJKLMNPQRSTUVWXYZ:/- (Letter 'O' and <space> is not included)
# _num_classes: 39
from module.ocr.al_ocr import AlOcr
return AlOcr(model_name='densenet-lite-gru', model_epoch=20, root='./bin/cnocr_models/azur_lane_jp',
name='azur_lane_jp')
return _ocr_class()(model_name='densenet-lite-gru', model_epoch=20,
root='./bin/cnocr_models/azur_lane_jp', name='azur_lane_jp')
@cached_property
def cnocr(self):
# Folder: ./bin/cnocr_models/cnocr
# Size: 9.51MB
# Size: 9.51MB (MXNet) / 9.75MB (ONNX)
# Model: densenet-lite-gru
# Epoch: 39
# Validation accuracy: 99.04%
# Font: Various
# Charset: Number, English character, Chinese character, symbols, <space>
# _num_classes: 6426
from module.ocr.al_ocr import AlOcr
return AlOcr(model_name='densenet-lite-gru', model_epoch=39, root='./bin/cnocr_models/cnocr', name='cnocr')
return _ocr_class()(model_name='densenet-lite-gru', model_epoch=39,
root='./bin/cnocr_models/cnocr', name='cnocr')
@cached_property
def jp(self):
from module.ocr.al_ocr import AlOcr
return AlOcr(model_name='densenet-lite-gru', model_epoch=125, root='./bin/cnocr_models/jp', name='jp')
# Folder: ./bin/cnocr_models/jp
# Size: 6.36MB (ONNX)
# Model: densenet-lite-gru
# Epoch: 125
return _ocr_class()(model_name='densenet-lite-gru', model_epoch=125,
root='./bin/cnocr_models/jp', name='jp')
@cached_property
def tw(self):
# Folder: ./bin/cnocr_models/tw
# Size: 8.43MB
# Size: 8.43MB (MXNet) / 8.64MB (ONNX)
# Model: densenet-lite-gru
# Epoch: 63
# Validation accuracy: 99.24%
# Font: Various, 6 kinds
# Charset: Numbers, Upper english characters, Chinese traditional characters
# _num_classes: 5322
from module.ocr.al_ocr import AlOcr
return AlOcr(model_name='densenet-lite-gru', model_epoch=63, root='./bin/cnocr_models/tw', name='tw')
return _ocr_class()(model_name='densenet-lite-gru', model_epoch=63,
root='./bin/cnocr_models/tw', name='tw')
OCR_MODEL = OcrModel()