Add models

app.py

@@ -115,30 +115,44 @@ with gr.Blocks(title='RapidOCR') as demo:
         - Python: 3.8
         - onnxruntime: 1.14.1
         - rapidocr_onnxruntime: 1.2.5""")
-    with gr.Box():
-        with gr.Row():
-            box_thresh = gr.Slider(minimum=0, maximum=1.0, value=0.5,
-                                   label='box_thresh', step=0.1,
-                                   interactive=True,
-                                   info='[0, 1.0]')
-            unclip_ratio = gr.Slider(minimum=1.5, maximum=2.0, value=1.6,
-                                     label='unclip_ratio', step=0.1,
-                                     interactive=True,
-                                     info='[1.5, 2.0]')
-            text_score = gr.Slider(minimum=0, maximum=1.0, value=0.5,
-                                   label='text_score', step=0.1,
-                                   interactive=True,
-                                   info='[0, 1.0]')
-        with gr.Row():
-            input_img = gr.Image(type='filepath', label='Input')
-            with gr.Column(scale=2):
-                out_img = gr.Image(type='filepath', label='Output_image')
-                out_txt = gr.outputs.Textbox(type='text', label='Output_text')
+    gr.Markdown('### 超参数调节')
+    with gr.Row():
+        box_thresh = gr.Slider(minimum=0, maximum=1.0, value=0.5,
+                               label='box_thresh', step=0.1,
+                               interactive=True,
+                               info='[0, 1.0]')
+        unclip_ratio = gr.Slider(minimum=1.5, maximum=2.0, value=1.6,
+                                 label='unclip_ratio', step=0.1,
+                                 interactive=True,
+                                 info='[1.5, 2.0]')
+        text_score = gr.Slider(minimum=0, maximum=1.0, value=0.5,
+                               label='text_score', step=0.1,
+                               interactive=True,
+                               info='[0, 1.0]')
+
+    gr.Markdown('### 模型选择')
+    with gr.Row():
+        text_det = gr.Dropdown(['ch_PP-OCRv3_det_infer.onnx',
+                                'ch_PP-OCRv2_det_infer.onnx',
+                                'ch_ppocr_server_v2.0_det_infer.onnx'],
+                               label='选择文本检测模型',
+                               value='ch_PP-OCRv3_det_infer.onnx',
+                               interactive=True)
+        text_rec = gr.Dropdown(['ch_PP-OCRv3_rec_infer.onnx',
+                                'ch_PP-OCRv2_rec_infer.onnx',
+                                'ch_ppocr_server_v2.0_rec_infer.onnx'],
+                               label='选择文本识别模型',
+                               value='ch_PP-OCRv3_rec_infer.onnx',
+                               interactive=True)
+
+    with gr.Row():
+        input_img = gr.Image(type='filepath', label='Input')
+        out_img = gr.Image(type='filepath', label='Output')
+    out_txt = gr.outputs.Textbox(type='text', label='RecText')
     button = gr.Button('Submit')
     button.click(fn=inference,
                  inputs=[input_img, box_thresh, unclip_ratio, text_score],
                  outputs=[out_img, out_txt])
-
     gr.Examples(examples=examples,
                 inputs=[input_img, box_thresh, unclip_ratio, text_score],
                 outputs=[out_img, out_txt], fn=inference)

models/text_det/ch_PP-OCRv2_det_infer.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:43d01120eab9f00e8d7541ad0f906dd9c7f28394233e47279073fab73f82da87
+size 2341867

models/text_det/ch_PP-OCRv3_det_infer.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3439588c030faea393a54515f51e983d8e155b19a2e8aba7891934c1cf0de526
+size 2432880

models/text_det/ch_ppocr_server_v2.0_det_infer.onnx

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

models/text_rec/ch_PP-OCRv2_rec_infer.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:86eec59d02644a8890d54f4aa5fe7402cf2848450fc7ac613d4677cd6025b3da
+size 8402009

models/text_rec/ch_PP-OCRv3_rec_infer.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:897a3ededb38fee0dae2c1ccee38241f37df202c9509e3abca02e9217c5ee615
+size 10690752

models/text_rec/ch_ppocr_server_v2.0_rec_infer.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9d0714a2b44f5d7ac7a17e44d0e1961393447590c895f4197f37e855e3464f53
+size 111590126