Loren commited on
Commit
6b54048
·
1 Parent(s): f5e8d20

Update app_pages/ocr_comparator.py

Browse files
Files changed (1) hide show
  1. app_pages/ocr_comparator.py +7 -7
app_pages/ocr_comparator.py CHANGED
@@ -2,12 +2,12 @@
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,7 +21,7 @@ from PIL import Image, ImageColor
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,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