mirror of
https://github.com/RapidAI/RapidOCR.git
synced 2026-11-12 05:50:49 +08:00
SWHLandCopilot
d48f31e815
* chore: update files * fix(rapidocr): fixed issue #381 * chore: update demo * Update python/setup.py Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> --------- Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
28 lines
832 B
Python
28 lines
832 B
Python
# -*- encoding: utf-8 -*-
|
|
# @Author: SWHL
|
|
# @Contact: liekkaskono@163.com
|
|
import shutil
|
|
from pathlib import Path
|
|
|
|
|
|
root_dir = Path(__file__).resolve().parent
|
|
DEFAULT_CFG_PATH = root_dir / "config.yaml"
|
|
|
|
|
|
def generate_cfg(args):
|
|
if args.save_cfg_file is None:
|
|
args.save_cfg_file = "./default_rapidocr.yaml"
|
|
|
|
shutil.copyfile(DEFAULT_CFG_PATH, args.save_cfg_file)
|
|
print(f"The config file has saved in {args.save_cfg_file}")
|
|
|
|
|
|
def check_install(ocr_engine):
|
|
img_url = "https://github.com/RapidAI/RapidOCR/blob/a9bb7c1f44b6e00556ada90ac588f020d7637c4b/python/tests/test_files/ch_en_num.jpg?raw=true"
|
|
result = ocr_engine(img_url)
|
|
|
|
if result.txts is None or result.txts[0] != "正品促销":
|
|
raise ValueError("The installation is incorrect!")
|
|
|
|
print("Success! rapidocr is installed correctly!")
|