提交 671cb14a 编写于 作者: T tink2123

modified default shape

上级 16752dd7
...@@ -407,7 +407,7 @@ def main(args): ...@@ -407,7 +407,7 @@ def main(args):
logger.info( logger.info(
"In PP-OCRv3, rec_image_shape parameter defaults to '3, 48, 320', " "In PP-OCRv3, rec_image_shape parameter defaults to '3, 48, 320', "
"if you are using an older PP-OCR, please set --rec_image_shape='3,32,320'" "if you are using recognition model with PP-OCRv2 or an older version, please set --rec_image_shape='3,32,320"
) )
# warmup 2 times # warmup 2 times
if args.warmup: if args.warmup:
......
...@@ -59,7 +59,7 @@ class TextSystem(object): ...@@ -59,7 +59,7 @@ class TextSystem(object):
for bno in range(bbox_num): for bno in range(bbox_num):
cv2.imwrite( cv2.imwrite(
os.path.join(output_dir, os.path.join(output_dir,
f"mg_crop_{bno+self.crop_image_res_index}.jpg "), f"mg_crop_{bno+self.crop_image_res_index}.jpg"),
img_crop_list[bno]) img_crop_list[bno])
logger.debug(f"{bno}, {rec_res[bno]}") logger.debug(f"{bno}, {rec_res[bno]}")
self.crop_image_res_index += bbox_num self.crop_image_res_index += bbox_num
...@@ -134,7 +134,7 @@ def main(args): ...@@ -134,7 +134,7 @@ def main(args):
save_results = [] save_results = []
logger.info("In PP-OCRv3, rec_image_shape parameter defaults to '3, 48, 320', " logger.info("In PP-OCRv3, rec_image_shape parameter defaults to '3, 48, 320', "
"if you are using an older PP-OCR, please set --rec_image_shape='3,32,320'") "if you are using recognition model with PP-OCRv2 or an older version, please set --rec_image_shape='3,32,320")
# warm up 10 times # warm up 10 times
if args.warmup: if args.warmup:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册