Skip to content

GitLab

Commit b6df9b18 authored by myhloli's avatar myhloli
Browse files

feat(cli): set "full" as default model_mode for better accuracy

parent 4d8d7ee4
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
magic_pdf/cli/magicpdf.py
View file @ b6df9b18
...@@ -180,7 +180,7 @@ def cli(): ...@@ -180,7 +180,7 @@ def cli():
default="auto", default="auto",
) )
@click.option("--inside_model", type=click.BOOL, default=False, help="使用内置模型测试") @click.option("--inside_model", type=click.BOOL, default=False, help="使用内置模型测试")
@click.option("--model_mode", type=click.STRING, default="lite", help="内置模型选择。lite: 快速解析，精度较低，full: 高精度解析，速度较慢") @click.option("--model_mode", type=click.STRING, default="full", help="内置模型选择。lite: 快速解析，精度较低，full: 高精度解析，速度较慢")
def json_command(json, method, inside_model, model_mode): def json_command(json, method, inside_model, model_mode):
model_config.__use_inside_model__ = inside_model model_config.__use_inside_model__ = inside_model
model_config.__model_mode__ = model_mode model_config.__model_mode__ = model_mode
...@@ -233,7 +233,7 @@ def json_command(json, method, inside_model, model_mode): ...@@ -233,7 +233,7 @@ def json_command(json, method, inside_model, model_mode):
default="auto", default="auto",
) )
@click.option("--inside_model", type=click.BOOL, default=False, help="使用内置模型测试") @click.option("--inside_model", type=click.BOOL, default=False, help="使用内置模型测试")
@click.option("--model_mode", type=click.STRING, default="lite", help="内置模型选择。lite: 快速解析，精度较低，full: 高精度解析，速度较慢") @click.option("--model_mode", type=click.STRING, default="full", help="内置模型选择。lite: 快速解析，精度较低，full: 高精度解析，速度较慢")
def local_json_command(local_json, method, inside_model, model_mode): def local_json_command(local_json, method, inside_model, model_mode):
model_config.__use_inside_model__ = inside_model model_config.__use_inside_model__ = inside_model
model_config.__model_mode__ = model_mode model_config.__model_mode__ = model_mode
...@@ -287,7 +287,7 @@ def local_json_command(local_json, method, inside_model, model_mode): ...@@ -287,7 +287,7 @@ def local_json_command(local_json, method, inside_model, model_mode):
default="auto", default="auto",
) )
@click.option("--inside_model", type=click.BOOL, default=False, help="使用内置模型测试") @click.option("--inside_model", type=click.BOOL, default=False, help="使用内置模型测试")
@click.option("--model_mode", type=click.STRING, default="lite", help="内置模型选择。lite: 快速解析，精度较低，full: 高精度解析，速度较慢") @click.option("--model_mode", type=click.STRING, default="full", help="内置模型选择。lite: 快速解析，精度较低，full: 高精度解析，速度较慢")
def pdf_command(pdf, model, method, inside_model, model_mode): def pdf_command(pdf, model, method, inside_model, model_mode):
model_config.__use_inside_model__ = inside_model model_config.__use_inside_model__ = inside_model
model_config.__model_mode__ = model_mode model_config.__model_mode__ = model_mode
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment