wai572 commited on
Commit
ad5b28d
·
1 Parent(s): 8a714a9
Files changed (1) hide show
  1. main.py +8 -2
main.py CHANGED
@@ -65,7 +65,13 @@ def load_ocr_model():
65
  global reader
66
 
67
  # 使用する言語と、モデルの保存先ディレクトリを指定してReaderを初期化
68
- reader = PaddleOCR(lang="en", use_angle_cls=False)
 
 
 
 
 
 
69
  print(f"PaddleOCR model loaded successfully.")
70
 
71
 
@@ -210,7 +216,7 @@ async def analyze_image(image_paths: list[UploadFile]):
210
  text_mask = get_not_white_mask(img)
211
  masked_img = cv2.bitwise_and(img, img, mask=text_mask)
212
  # result = reader.readtext(candidate["img"])
213
- result = reader.recognize([masked_img])
214
  print(result)
215
  if len(result) > 0:
216
  ocr_results.append(result)
 
65
  global reader
66
 
67
  # 使用する言語と、モデルの保存先ディレクトリを指定してReaderを初期化
68
+ reader = PaddleOCR(
69
+ lang="en",
70
+ use_angle_cls=False,
71
+ use_doc_orientation_classify=False,
72
+ use_doc_unwarping=False,
73
+ use_textline_orientation=False,
74
+ )
75
  print(f"PaddleOCR model loaded successfully.")
76
 
77
 
 
216
  text_mask = get_not_white_mask(img)
217
  masked_img = cv2.bitwise_and(img, img, mask=text_mask)
218
  # result = reader.readtext(candidate["img"])
219
+ result = reader.predict(masked_img)
220
  print(result)
221
  if len(result) > 0:
222
  ocr_results.append(result)