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

Fix python/rapidOCR parameter error

Browse Source
This commit is contained in:
SWHL committed 2021-11-25 19:23:23 +08:00
1 parent 08022053c5
commit 686e4ceca1
2 files changed
+11 -7
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

python/rapidOCR.py
+9 -5
View File
@@ -89,6 +89,7 @@ def draw_ocr_box_txt(image, boxes, txts,
color = (random.randint(0, 255),
random.randint(0, 255),
random.randint(0, 255))
print(box)
draw_left.polygon(box, fill=color)
draw_right.polygon([box[0][0], box[0][1],
box[1][0], box[1][1],
@@ -96,10 +97,10 @@ def draw_ocr_box_txt(image, boxes, txts,
box[3][0], box[3][1]],
outline=color)
box_height = math.sqrt((box[0][0] - box[3][0])**2 \
box_height = math.sqrt((box[0][0] - box[3][0])**2
+ (box[0][1] - box[3][1])**2)
box_width = math.sqrt((box[0][0] - box[1][0])**2 \
box_width = math.sqrt((box[0][0] - box[1][0])**2
+ (box[0][1] - box[1][1])**2)
if box_height > 2 * box_width:
@@ -267,8 +268,8 @@ if __name__ == '__main__':
parser.add_argument('--image_path', type=str,
default='test_images/det_images/')
parser.add_argument('--text_score', type=float, default=0.5)
parser.add_argument('--keys_path', type=str,
parser.add_argument('--keys_path', type=str,
default="ch_ppocr_mobile_v2_rec/ppocr_keys_v1.txt")
args = parser.parse_args()
text_score = args.text_score
@@ -283,7 +284,10 @@ if __name__ == '__main__':
if args.rec_model_path.find('server') != -1:
from ch_ppocr_server_v2_rec import TextRecognizer
else:
from ch_ppocr_mobile_v2_rec import TextRecognizer
if args.rec_model_path.find('en') != -1:
from en_number_ppocr_mobile_v2_rec import TextRecognizer
else:
from ch_ppocr_mobile_v2_rec import TextRecognizer
text_sys = TextSystem(args.det_model_path,
args.rec_model_path,
python/rapidOCR.sh
+2 -2
View File
@@ -3,10 +3,10 @@
# The img_path can be file path or folder path.
img_path="test_images/det_images/1.jpg"
key_path="ch_ppocr_mobile_v2_rec/ppocr_keys_v1.txt"
keys_path="ch_ppocr_mobile_v2_rec/ppocr_keys_v1.txt"
python rapidOCR.py --det_model_path models/ch_ppocr_mobile_v2.0_det_infer.onnx \
--cls_model_path models/ch_ppocr_mobile_v2.0_cls_infer.onnx \
--rec_model_path models/ch_ppocr_mobile_v2.0_rec_infer.onnx \
--image_path ${img_path} \
--text_score 0.5 \
--key_path ${key_path}
--keys_path ${keys_path}