MMCV version 1.7.0

pages/App.py

@@ -496,7 +496,7 @@ def process_detect(in_image_path, _in_list_images, _in_list_readers, in_list_par
     return out_list_images, out_list_coordinates
 
 ###
-def draw_detected(in_image, in_boxes_coordinates, in_color, posit=None, in_thickness=4):
+def draw_detected(in_image, in_boxes_coordinates, in_color, posit='None', in_thickness=4):
     """Draw boxes around detected text
 
     Args:
@@ -517,7 +517,7 @@ def draw_detected(in_image, in_boxes_coordinates, in_color, posit=None, in_thick
     for ind_box, box in enumerate(in_boxes_coordinates):
         box = np.reshape(np.array(box), [-1, 1, 2]).astype(np.int64)
         work_img = cv2.polylines(np.array(work_img), [box], True, in_color, in_thickness)
-        if posit != None:
+        if posit != 'None':
             if posit == 'top_left':
                 pos = tuple(box[0][0])
             elif posit == 'top_right':
@@ -903,7 +903,7 @@ def highlight():
 
 ##----------- Initializations ---------------------------------------------------------------------
 #print("PID : ", os.getpid())
-
+st.set_page_config(layout="wide")
 st.title("OCR solutions comparator")
 st.markdown("##### *EasyOCR, PPOCR, MMOCR, Tesseract*")
 #st.markdown("#### PID : " + str(os.getpid()))

requirements.txt

@@ -2,13 +2,15 @@
 easyocr
 streamlit
 opencv-python-headless==4.5.5.64
+torch==1.12.1
+torchvision==0.13.1
 Pillow
+mmcv-full --no-binary mmcv-full
 mmdet==2.25.1
-mmocr==0.6.1
+mmocr
 paddleocr==2.6
 paddlepaddle==2.3.2
 mycolorpy==1.5.1
 plotly==5.10.0
 plotly-express==0.4.1
-mmcv-full==1.6.1 --no-binary mmcv-full==1.6.1
 pytesseract==0.3.10