"tests/vscode:/vscode.git/clone" did not exist on "8a227bfa03b911fff681c8559011a2ac252bd24e"
Commit 399023f5 authored by myhloli's avatar myhloli
Browse files

fix: update default value of MINERU_MIN_BATCH_INFERENCE_SIZE to 128 in pipeline_analyze.py

parent b420094b
...@@ -75,9 +75,9 @@ def doc_analyze( ...@@ -75,9 +75,9 @@ def doc_analyze(
): ):
""" """
适当调大MIN_BATCH_INFERENCE_SIZE可以提高性能,可能会增加显存使用量, 适当调大MIN_BATCH_INFERENCE_SIZE可以提高性能,可能会增加显存使用量,
可通过环境变量MINERU_MIN_BATCH_INFERENCE_SIZE设置,默认值为100 可通过环境变量MINERU_MIN_BATCH_INFERENCE_SIZE设置,默认值为128
""" """
min_batch_inference_size = int(os.environ.get('MINERU_MIN_BATCH_INFERENCE_SIZE', 100)) min_batch_inference_size = int(os.environ.get('MINERU_MIN_BATCH_INFERENCE_SIZE', 128))
# 收集所有页面信息 # 收集所有页面信息
all_pages_info = [] # 存储(dataset_index, page_index, img, ocr, lang, width, height) all_pages_info = [] # 存储(dataset_index, page_index, img, ocr, lang, width, height)
......
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