diff --git a/tools/infer/predict_rec.py b/tools/infer/predict_rec.py index 340ea3328fa83e66953c7b1a70f5e76290e32199..3664ef2caf4b888d6a3918202256c99cc54c5eb1 100755 --- a/tools/infer/predict_rec.py +++ b/tools/infer/predict_rec.py @@ -407,7 +407,7 @@ def main(args): 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" ) # warmup 2 times if args.warmup: diff --git a/tools/infer/predict_system.py b/tools/infer/predict_system.py index 7e09c66d564d0179c913ceaea3a6ebfdfe079dd1..625d365f45c578d051974d7174e26246e9bc2442 100755 --- a/tools/infer/predict_system.py +++ b/tools/infer/predict_system.py @@ -59,7 +59,7 @@ class TextSystem(object): for bno in range(bbox_num): cv2.imwrite( 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]) logger.debug(f"{bno}, {rec_res[bno]}") self.crop_image_res_index += bbox_num @@ -134,7 +134,7 @@ def main(args): save_results = [] 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 if args.warmup: