sui_feng/AzurLaneAutoScript
1
0
Fork 0
mirror of https://gitee.com/sui-feng-cb/AzurLaneAutoScript1 synced 2026-03-12 00:28:20 +08:00
Code Activity

Opt: Patch pkg_resources for faster startup

Browse Source
This commit is contained in:
LmeSzinc
2024-04-15 18:43:42 +08:00
parent 31da8d42a1
commit 4858d55c46
3 changed files with 93 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
module/ocr/al_ocr.py
View File

@@ -13,16 +13,14 @@ from cnocr.cn_ocr import (check_model_name, data_dir, gen_network, load_module,
read_charset)
from cnocr.fit.ctc_metrics import CtcMetrics
from cnocr.hyperparams.cn_hyperparams import CnHyperparams as Hyperparams
from module.device.pkg_resources import PACKAGE_CACHE
def get_mxnet_context():
import re
import cnocr
site_packages = os.path.abspath(os.path.join(cnocr.__file__, '../../'))
for file in os.listdir(site_packages):
# mxnet_cu101-1.6.0.dist-info
if re.match(r'^mxnet[-_]cu\d+', file):
logger.info(f'MXNet gpu package: {file} found, using it')
for dist in PACKAGE_CACHE.dict_installed_packages.values():
# mxnet_cu101
if dist.dist.startswith('mxnet_cu'):
logger.info(f'MXNet gpu package: {dist.dist}=={dist.version} found, using it')
return 'gpu'
return 'cpu'