Skip to content

Commit c66577d

Browse files
committed
update log
1 parent e37521d commit c66577d

3 files changed

Lines changed: 6 additions & 3 deletions

File tree

agentocr/infer/utility.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -121,6 +121,9 @@ def get_vis_font(vis_font_path):
121121
def str2providers(str, logger):
122122
available_providers = ort.get_available_providers()
123123

124+
if str.lower() == 'auto':
125+
return available_providers
126+
124127
providers_dict = {
125128
provider.lower(): provider
126129
for provider in available_providers
@@ -148,7 +151,7 @@ def init_args():
148151
parser = argparse.ArgumentParser()
149152

150153
# params for onnx engine
151-
parser.add_argument("--providers", type=str, default='')
154+
parser.add_argument("--providers", type=str, default='auto')
152155

153156
# params for text detector
154157
parser.add_argument("--det_algorithm", type=str, default='DB')

docs/package_usage.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -227,7 +227,7 @@
227227
### 3.2 参数说明
228228
| 字段 | 说明 | 默认值 |
229229
|-------------------------|----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|-------------------------|
230-
| providers | 使用的计算后端,使用字符串设定使用顺序,使用逗号分隔,不区分大小写,如 “cuda,cpu”,默认空字符串,为所有可用后端自动选择(可能并非最佳顺序) | |
230+
| providers | 使用的计算后端,使用字符串设定使用顺序,使用逗号分隔,不区分大小写,如 “cuda,cpu”,默认自动选择所有可用后端(可能并非最佳顺序) | auto |
231231
| det_algorithm | 使用的检测算法类型 | DB |
232232
| det_model_dir | 检测模型文件,文件名或文件路径 | ch_ppocr_mobile_v2.0_det |
233233
| det_max_side_len | 检测算法前向时图片长边的最大尺寸,当长边超出这个值时会将长边resize到这个大小,短边等比例缩放 | 960 |

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ def readme():
1414
],
1515
include_package_data=True,
1616
entry_points={"console_scripts": ["agentocr = agentocr:command"]},
17-
version='1.1.1',
17+
version='1.1.2',
1818
install_requires=[
1919
'shapely', 'pyclipper', 'numpy', 'opencv-python', 'pillow', 'wget'
2020
],

0 commit comments

Comments
 (0)