mtfyun/rapidocr
1
0 Fork 0
mirror of https://github.com/RapidAI/RapidOCR.git synced 2026-11-10 21:10:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update use_cuda parameter in config.yaml

Browse Source
This commit is contained in:
SWHL committed 2024-03-04 22:59:40 +08:00
1 parent 2f66907a6c
commit bd2eba459f
2 files changed
+6 -4
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

python/demo.py
+1 -1
View File
@@ -12,7 +12,7 @@ from rapidocr_onnxruntime import RapidOCR, VisRes
engine = RapidOCR()
vis = VisRes()
image_path = "tests/test_files/test_without_det.png"
image_path = "tests/test_files/test_without_det.jpg"
with open(image_path, "rb") as f:
img = f.read()
python/rapidocr_onnxruntime/config.yaml
+5 -3
View File
@@ -7,6 +7,8 @@ Global:
min_height: 30
width_height_ratio: 8
use_cuda: &use_cuda false
intra_op_num_threads: &intra_nums -1
inter_op_num_threads: &inter_nums -1
@@ -14,7 +16,7 @@ Det:
intra_op_num_threads: *intra_nums
inter_op_num_threads: *inter_nums
use_cuda: false
use_cuda: *use_cuda
model_path: models/ch_PP-OCRv4_det_infer.onnx
@@ -32,7 +34,7 @@ Cls:
intra_op_num_threads: *intra_nums
inter_op_num_threads: *inter_nums
use_cuda: false
use_cuda: *use_cuda
model_path: models/ch_ppocr_mobile_v2.0_cls_infer.onnx
@@ -45,7 +47,7 @@ Rec:
intra_op_num_threads: *intra_nums
inter_op_num_threads: *inter_nums
use_cuda: false
use_cuda: *use_cuda
model_path: models/ch_PP-OCRv4_rec_infer.onnx