diff --git a/tools/infer/predict_rec.py b/tools/infer/predict_rec.py index f3256be..c8a82df 100644 --- a/tools/infer/predict_rec.py +++ b/tools/infer/predict_rec.py @@ -11,7 +11,7 @@ import time import torch from pytorchocr.base_ocr_v20 import BaseOCRV20 -import tools.infer.pytorchocr_utility as utility +import pytorchocr_utility as utility from pytorchocr.postprocess import build_post_process from pytorchocr.utils.utility import get_image_file_list, check_and_read_gif @@ -388,4 +388,4 @@ def main(args): if __name__ == '__main__': - main(utility.parse_args()) \ No newline at end of file + main(utility.parse_args())