Loren commited on
Commit
0a84764
1 Parent(s): 6c40532

Add extra demo file

Browse files
Files changed (5) hide show
  1. img.jpg +0 -0
  2. img_demo_1.jpg +0 -0
  3. img_demo_2.jpg +0 -0
  4. img_demo_2.png +0 -0
  5. pages/App.py +14 -7
img.jpg CHANGED
img_demo_1.jpg CHANGED
img_demo_2.jpg ADDED
img_demo_2.png DELETED
Binary file (45.6 kB)
 
pages/App.py CHANGED
@@ -443,7 +443,7 @@ def process_detect(in_image_path, _in_list_images, _in_list_readers, in_list_par
443
  # Visualization
444
  if easyocr_boxes_coordinates:
445
  easyocr_image_detect = draw_detected(_in_list_images[0], easyocr_boxes_coordinates, \
446
- in_color, 'None', 7)
447
  else:
448
  easyocr_boxes_coordinates = easyocr_status
449
  ##
@@ -824,7 +824,7 @@ def draw_reco_images(in_image, in_boxes_coordinates, in_list_texts, in_list_conf
824
  cv2.putText(list_reco_images[ind_tessocr], \
825
  in_df_results_tesseract.iloc[num]['text'], \
826
  (box[0][0],int(np.round((box[0][1]+box[2][1])/2,0))), \
827
- cv2.FONT_HERSHEY_DUPLEX, in_font_scale, text_color, 4)
828
 
829
  with show_reco.container():
830
  # Draw the results, 2 images per line
@@ -852,6 +852,7 @@ def draw_reco_images(in_image, in_boxes_coordinates, in_list_texts, in_list_conf
852
  cols[ind_col].write(list_reco_status[ind], \
853
  use_column_width=True)
854
 
 
855
  ###
856
  def highlight():
857
  """Draw recognized text on original image, for each OCR solution used
@@ -901,7 +902,7 @@ def get_demo():
901
  """
902
 
903
  out_img_demo_1 = Image.open("img_demo_1.jpg")
904
- out_img_demo_2 = Image.open("img_demo_2.png")
905
 
906
  return out_img_demo_1, out_img_demo_2
907
 
@@ -911,7 +912,13 @@ def get_demo():
911
 
912
  ##----------- Initializations ---------------------------------------------------------------------
913
  #print("PID : ", os.getpid())
914
- st.set_page_config(layout="wide")
 
 
 
 
 
 
915
  st.title("OCR solutions comparator")
916
  st.markdown("##### *EasyOCR, PPOCR, MMOCR, Tesseract*")
917
  #st.markdown("#### PID : " + str(os.getpid()))
@@ -952,7 +959,7 @@ if img_typ == 'Use a demo file':
952
  if demo_used == 'File 1':
953
  image_file = 'img_demo_1.jpg'
954
  else:
955
- image_file = 'img_demo_2.png'
956
 
957
  ##----------- Process input image -----------------------------------------------------------------
958
  if image_file is not None:
@@ -1168,7 +1175,7 @@ Use rectlar box to calculate faster, and polygonal box more accurate for curved
1168
  horizontal=True, on_change=highlight)
1169
 
1170
  ##----------- Form with hyperparameters for recognition -----------------------
1171
- st.markdown("##### Hyperparameters values for recognition")
1172
  with st.form("form2"):
1173
  with st.expander("Choose recognition hyperparameters for " + reader_type_list[0], \
1174
  expanded=False):
@@ -1341,7 +1348,7 @@ Use rectlar box to calculate faster, and polygonal box more accurate for curved
1341
  in_conf_threshold=st.session_state.conf_threshold_sld)
1342
 
1343
  st.subheader("Recognition details")
1344
- with st.expander("Detailed areas for EasyOCR, PPOCR, MMOCR", expanded=False):
1345
  cols = st.columns(cols_size)
1346
  cols[0].markdown('#### Detected area')
1347
  for i in range(1, (len(reader_type_list)-1)*2, 2):
 
443
  # Visualization
444
  if easyocr_boxes_coordinates:
445
  easyocr_image_detect = draw_detected(_in_list_images[0], easyocr_boxes_coordinates, \
446
+ in_color, 'None', 3)
447
  else:
448
  easyocr_boxes_coordinates = easyocr_status
449
  ##
 
824
  cv2.putText(list_reco_images[ind_tessocr], \
825
  in_df_results_tesseract.iloc[num]['text'], \
826
  (box[0][0],int(np.round((box[0][1]+box[2][1])/2,0))), \
827
+ cv2.FONT_HERSHEY_DUPLEX, in_font_scale, text_color, 2)
828
 
829
  with show_reco.container():
830
  # Draw the results, 2 images per line
 
852
  cols[ind_col].write(list_reco_status[ind], \
853
  use_column_width=True)
854
 
855
+ st.markdown(' 💡 Wrong font size? you can adjust it below:')
856
  ###
857
  def highlight():
858
  """Draw recognized text on original image, for each OCR solution used
 
902
  """
903
 
904
  out_img_demo_1 = Image.open("img_demo_1.jpg")
905
+ out_img_demo_2 = Image.open("img_demo_2.jpg")
906
 
907
  return out_img_demo_1, out_img_demo_2
908
 
 
912
 
913
  ##----------- Initializations ---------------------------------------------------------------------
914
  #print("PID : ", os.getpid())
915
+ st.set_page_config(page_title='OCR Comparator', layout ="wide")
916
+ st.markdown(f'''
917
+ <style>
918
+ section[data-testid="stSidebar"] .css-ng1t4o {{width: 14rem;}}
919
+ section[data-testid="stSidebar"] .css-1d391kg {{width: 14rem;}}
920
+ </style>
921
+ ''',unsafe_allow_html=True)
922
  st.title("OCR solutions comparator")
923
  st.markdown("##### *EasyOCR, PPOCR, MMOCR, Tesseract*")
924
  #st.markdown("#### PID : " + str(os.getpid()))
 
959
  if demo_used == 'File 1':
960
  image_file = 'img_demo_1.jpg'
961
  else:
962
+ image_file = 'img_demo_2.jpg'
963
 
964
  ##----------- Process input image -----------------------------------------------------------------
965
  if image_file is not None:
 
1175
  horizontal=True, on_change=highlight)
1176
 
1177
  ##----------- Form with hyperparameters for recognition -----------------------
1178
+ st.markdown("##### Hyperparameters values for recognition:")
1179
  with st.form("form2"):
1180
  with st.expander("Choose recognition hyperparameters for " + reader_type_list[0], \
1181
  expanded=False):
 
1348
  in_conf_threshold=st.session_state.conf_threshold_sld)
1349
 
1350
  st.subheader("Recognition details")
1351
+ with st.expander("Detailed areas for EasyOCR, PPOCR, MMOCR", expanded=True):
1352
  cols = st.columns(cols_size)
1353
  cols[0].markdown('#### Detected area')
1354
  for i in range(1, (len(reader_type_list)-1)*2, 2):