mirror of
https://github.com/RapidAI/RapidOCR.git
synced 2026-11-10 21:10:55 +08:00
25 lines
542 B
Python
25 lines
542 B
Python
# -*- encoding: utf-8 -*-
|
|
# @Author: SWHL
|
|
# @Contact: liekkaskono@163.com
|
|
import cv2
|
|
|
|
from rapidocr_onnxruntime import RapidOCR, VisRes
|
|
|
|
# from rapidocr_openvino import RapidOCR, VisRes
|
|
|
|
|
|
engine = RapidOCR()
|
|
vis = VisRes(font_path="resources/fonts/FZYTK.TTF")
|
|
|
|
image_path = "tests/test_files/ch_en_num.jpg"
|
|
with open(image_path, "rb") as f:
|
|
img = f.read()
|
|
|
|
result, elapse_list = engine(img)
|
|
print(result)
|
|
print(elapse_list)
|
|
|
|
boxes, txts, scores = list(zip(*result))
|
|
vis_img = vis(img, boxes, txts, scores)
|
|
cv2.imwrite("vis.png", vis_img)
|