You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
22 lines
865 B
Python
22 lines
865 B
Python
import os
|
|
import sys
|
|
|
|
_MODEL_VERSION_ = "20230229"
|
|
|
|
if 'win32' in sys.platform:
|
|
_DEFAULT_FOLDER_ = os.path.join(os.environ['HOMEPATH'], ".hyperlpr3")
|
|
else:
|
|
_DEFAULT_FOLDER_ = os.path.join(os.environ['HOME'], ".hyperlpr3")
|
|
|
|
_ONLINE_URL_ = "https://tunm.oss-cn-hangzhou.aliyuncs.com/hyperlpr3/"
|
|
|
|
onnx_runtime_config = dict(
|
|
det_model_path_320x=os.path.join(_MODEL_VERSION_, "onnx", "y5fu_320x_sim.onnx"),
|
|
det_model_path_640x=os.path.join(_MODEL_VERSION_, "onnx", "y5fu_640x_sim.onnx"),
|
|
rec_model_path=os.path.join(_MODEL_VERSION_, "onnx", "rpv3_mdict_160_r3.onnx"),
|
|
cls_model_path=os.path.join(_MODEL_VERSION_, "onnx", "litemodel_cls_96x_r1.onnx"),
|
|
)
|
|
|
|
onnx_model_maps = ["det_model_path_320x", "det_model_path_640x", "rec_model_path", "cls_model_path"]
|
|
|
|
_REMOTE_URL_ = "https://github.com/szad670401/HyperLPR/blob/master/resource/models/onnx/" |