mirror of
https://github.com/PaddlePaddle/PaddleOCR.git
synced 2025-07-05 16:10:17 +00:00
3d03ca5500
* Init new paddleocr * Remove unused dependency * Fix typos * Fix * Add doc understanding modules * Fix package finding * Normalize name * Fix setting bugs * Fix setting bug * Support single model inference * Add PP-ChatOCRv4-doc * Add pp_chatocrv4_doc tests * Enable MKL-DNN when available * add seal_text_detection modules * add layout_detection and table_cells_detection modules * add testing scripts * Fix desc * add text_image_unwarping and table_structure_recognition modules * add formula_recognition and doc_vlm modules * update formula_recognition default_model_name * add MKLDNN_BLOCKLIST * update MKLDNN log * add seal rec pipeline * fix sth * fix sth * add doc preprocessor pipeline * fix sth * add doc understanding * add table_rec_v2, ppstructurev3, formula_rec pipelines * move test files * forward kwargs to pipeline.predict * clean test files * Add missing kwargs * Fix typo * Fix typo * rerun CI * update mkldnn BLOCKLIST * update * update warning message * fix cli args * update PIPELINE_MKLDNN_BLOCKLIST * update of workflow * skip resource_intensive tests * update config * skip ppdocbee test_predict_params --------- Co-authored-by: zhangyue66 <zhangyue66@baidu.com> Co-authored-by: zhangzelun <zhangzelun@baidu.com>
51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
# Copyright (c) 2025 PaddlePaddle Authors. All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
|
|
def str2bool(v, /):
|
|
return v.lower() in ("true", "yes", "t", "y", "1")
|
|
|
|
|
|
def get_subcommand_args(args):
|
|
args = vars(args).copy()
|
|
args.pop("subcommand")
|
|
args.pop("executor")
|
|
return args
|
|
|
|
|
|
def add_simple_inference_args(subparser):
|
|
subparser.add_argument(
|
|
"-i", "--input", type=str, required=True, help="Input path or URL."
|
|
)
|
|
subparser.add_argument(
|
|
"--save_path",
|
|
type=str,
|
|
default="output",
|
|
help="Path to the output directory.",
|
|
)
|
|
|
|
|
|
def perform_simple_inference(wrapper_cls, params):
|
|
input_ = params.pop("input")
|
|
save_path = params.pop("save_path")
|
|
|
|
wrapper = wrapper_cls(**params)
|
|
|
|
result = wrapper.predict(input_)
|
|
|
|
for res in result:
|
|
res.print()
|
|
if save_path:
|
|
res.save_all(save_path)
|