mirror of
https://github.com/PaddlePaddle/PaddleOCR.git
synced 2025-07-05 08:00:53 +00:00
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)
|