Spaces:
Sleeping
Sleeping
correct
Browse files
main.py
CHANGED
@@ -65,7 +65,13 @@ def load_ocr_model():
|
|
65 |
global reader
|
66 |
|
67 |
# 使用する言語と、モデルの保存先ディレクトリを指定してReaderを初期化
|
68 |
-
reader = PaddleOCR(
|
|
|
|
|
|
|
|
|
|
|
|
|
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.
|
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)
|