Update app_pages/ocr_comparator.py
Browse files
app_pages/ocr_comparator.py
CHANGED
@@ -2,12 +2,12 @@
|
|
2 |
EasyOcr, PaddleOCR, MMOCR, Tesseract
|
3 |
"""
|
4 |
|
5 |
-
|
6 |
#
|
7 |
-
|
8 |
-
|
9 |
-
|
10 |
-
|
11 |
|
12 |
import streamlit as st
|
13 |
import plotly.express as px
|
@@ -21,7 +21,7 @@ from PIL import Image, ImageColor
|
|
21 |
import PIL
|
22 |
import easyocr
|
23 |
from paddleocr import PaddleOCR
|
24 |
-
|
25 |
import pytesseract
|
26 |
from pytesseract import Output
|
27 |
import os
|
@@ -477,7 +477,7 @@ def app():
|
|
477 |
ppocr_boxes_coordinates, ppocr_status = ppocr_detect(_in_list_readers[1], in_image_path)
|
478 |
# Visualization
|
479 |
if ppocr_boxes_coordinates:
|
480 |
-
ppocr_image_detect = draw_detected(_in_list_images[0], ppocr_boxes_coordinates, \
|
481 |
in_color, 'None', 3)
|
482 |
else:
|
483 |
ppocr_image_detect = ppocr_status
|
|
|
2 |
EasyOcr, PaddleOCR, MMOCR, Tesseract
|
3 |
"""
|
4 |
|
5 |
+
import mim
|
6 |
#
|
7 |
+
mim.install(['mmengine>=0.7.1,<1.1.0'])
|
8 |
+
mim.install(['mmcv>=2.0.0rc4,<2.1.0'])
|
9 |
+
mim.install(['mmdet>=3.0.rc5,<3.2.0'])
|
10 |
+
mim.install(['mmocr'])
|
11 |
|
12 |
import streamlit as st
|
13 |
import plotly.express as px
|
|
|
21 |
import PIL
|
22 |
import easyocr
|
23 |
from paddleocr import PaddleOCR
|
24 |
+
from mmocr.utils.ocr import MMOCR
|
25 |
import pytesseract
|
26 |
from pytesseract import Output
|
27 |
import os
|
|
|
477 |
ppocr_boxes_coordinates, ppocr_status = ppocr_detect(_in_list_readers[1], in_image_path)
|
478 |
# Visualization
|
479 |
if ppocr_boxes_coordinates:
|
480 |
+
ppocr_image_detect = draw_detected(_in_list_images[0], ppocr_boxes_coordinates[0], \
|
481 |
in_color, 'None', 3)
|
482 |
else:
|
483 |
ppocr_image_detect = ppocr_status
|