yujunjun/PaddleOCR
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2025-11-18 02:53:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add dep installation CLI command (#15103)

Browse Source
This commit is contained in:
Lin Manhui 2025-05-18 21:12:31 +08:00 committed by GitHub
parent 2357c63a9a
commit a4fdd4dbdb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
paddleocr/_cli.py
View File

@ -13,6 +13,8 @@
# limitations under the License. # limitations under the License.
import argparse import argparse
import subprocess
import sys
import warnings import warnings
from ._models import ( from ._models import (
@ -79,12 +81,27 @@ def _register_models(subparsers):
subparser.set_defaults(executor=subcommand_executor.execute_with_args) subparser.set_defaults(executor=subcommand_executor.execute_with_args)
def _register_install_hpi_deps_command(subparsers):
def _install_hpi_deps(args):
hpip = f"hpi-{args.variant}"
try:
subprocess.check_call(["paddlex", "--install", hpip])
subprocess.check_call(["paddlex", "--install", "paddle2onnx"])
except subprocess.CalledProcessError:
sys.exit("Failed to install dependencies")
subparser = subparsers.add_parser("install_hpi_deps")
subparser.add_argument("variant", type=str, choices=["cpu", "gpu", "npu"])
subparser.set_defaults(executor=_install_hpi_deps)
def _parse_args(): def _parse_args():
parser = argparse.ArgumentParser(prog="paddleocr") parser = argparse.ArgumentParser(prog="paddleocr")
parser.add_argument("--version", action="version", version=f"%(prog)s {version}") parser.add_argument("--version", action="version", version=f"%(prog)s {version}")
subparsers = parser.add_subparsers(dest="subcommand") subparsers = parser.add_subparsers(dest="subcommand")
_register_pipelines(subparsers) _register_pipelines(subparsers)
_register_models(subparsers) _register_models(subparsers)
_register_install_hpi_deps_command(subparsers)
return parser.parse_args() return parser.parse_args()