Spaces:
Sleeping
Sleeping
tr
Browse files
main.py
CHANGED
@@ -65,7 +65,7 @@ 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 +210,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.ocr(masked_img)
|
214 |
print(result)
|
215 |
if len(result) > 0:
|
216 |
ocr_results.append(result)
|
|
|
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 |
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.ocr(masked_img, det=False)
|
214 |
print(result)
|
215 |
if len(result) > 0:
|
216 |
ocr_results.append(result)
|