wip

.gitattributes

@@ -64,3 +64,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.pdiparams filter=lfs diff=lfs merge=lfs -text
 *.info filter=lfs diff=lfs merge=lfs -text
 *.pdmodel filter=lfs diff=lfs merge=lfs -text
+*.ttf filter=lfs diff=lfs merge=lfs -text

app.py

@@ -16,7 +16,7 @@ uploaded_file = st.file_uploader("**Upload a Chest X-Ray Image**", type= ['png',
 if uploaded_file is not None:
     # Get actual image file
     fpath= get_image_path(uploaded_file)
-    st.image(fpath)
+    # st.image(fpath)
     res = start_ocr({
     'image_dir': fpath,
     'drop_score': 0.3,
@@ -37,4 +37,4 @@ if uploaded_file is not None:
     "show_log": True,
 })
 
-    st.write(json.dumps(res))
+    st.image(draw_ocr_results(fpath, res, './fonts/simfang.ttf'))

fonts/simfang.ttf

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:521c6f7546b4eb64fa4b0cd604bbd36333a20a57e388c8e2ad2ad07b9e593864
+size 10576012

utils.py

@@ -88,10 +88,11 @@ def draw_ocr_results(image_fp: Union[str, Path, Image.Image], ocr_outs, font_pat
 
     # cv2.imwrite(out_draw_fp, draw_img[:, :, ::-1])
 
-    plt.figure(figsize=(draw_img.shape[0]/100, draw_img.shape[1]/100), dpi=100)
-    plt.imshow(draw_img)
-    plt.axis('off')  # Turn off axis numbers
-    plt.show()
+    #plt.figure(figsize=(draw_img.shape[0]/100, draw_img.shape[1]/100), dpi=100)
+    #plt.imshow(draw_img)
+    #plt.axis('off')  # Turn off axis numbers
+    #plt.show()
+    return draw_img
 
 class MyObject:
     def __init__(self, dictionary):