move the paddle ocr in to the method

Cackbone · Apr 2, 2022 · 4557d00 · 4557d00
1 parent bb96f3f
commit 4557d00
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 11 deletions.
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
diff --git a/detect_text/text_detection.py b/detect_text/text_detection.py
@@ -6,7 +6,6 @@
 import time
 import os
 from os.path import join as pjoin
-from paddleocr import PaddleOCR
 
 
 def save_detection_json(file_path, texts, img_shape):
@@ -146,6 +145,8 @@ def text_detection(input_file='../data/input/30800.jpg', output_file='../data/ou
         texts = text_filter_noise(texts)
         texts = text_sentences_recognition(texts)
     elif method == 'paddle':
+        # The import of the paddle ocr can be separate to the beginning of the program if you decide to use this method
+        from paddleocr import PaddleOCR
         print('*** Detect Text through Paddle OCR ***')
         if paddle_model is None:
             paddle_model = PaddleOCR(use_angle_cls=True, lang="ch")