add examples

README.md

@@ -1,5 +1,5 @@
 ---
-title: CnOCR Demo
+title: Cn/En OCR Demo
 emoji: 🅞🅒🅡
 colorFrom: indigo
 colorTo: yellow
@@ -10,7 +10,7 @@ pinned: false
 license: apache-2.0
 ---
 
-# CnOCR (Cn-OCR)
+# CnOCR
 
 [**CnOCR**](https://github.com/breezedeus/cnocr)  is an **Optical Character Recognition (OCR)** toolkit for **Python 3**. It supports recognition of common characters in **English and numbers**, **Simplified Chinese**, **Traditional Chinese** (some models), and **vertical text** recognition. It comes with [**20+ well-trained models**](https://cnocr.readthedocs.io/zh/latest/models/) for different application scenarios and can be used directly after installation. Also, CnOCR provides simple training [commands](https://cnocr.readthedocs.io/zh/latest/train/) for users to train their own models. Welcome to join the WeChat contact group.
 

app.py

@@ -16,8 +16,11 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
+# Ref: https://huggingface.co/spaces/hysts/Manga-OCR/blob/main/app.py
 
 import os
+import json
+import functools
 
 import gradio as gr
 import cv2
@@ -29,6 +32,7 @@ from cnocr.utils import set_logger, draw_ocr_results, download
 
 
 logger = set_logger()
+MODELS = {}
 
 
 def plot_for_debugging(rotated_img, one_out, box_score_thresh, crop_ncols, prefix_fp):
@@ -64,15 +68,24 @@ def plot_for_debugging(rotated_img, one_out, box_score_thresh, crop_ncols, prefi
 
 
 def get_ocr_model(det_model_name, rec_model_name, det_more_configs):
+    global MODELS
+    config_str = json.dumps(det_more_configs)
+    if (det_model_name, rec_model_name, config_str) in MODELS:
+        return MODELS[(det_model_name, rec_model_name, config_str)]
+
     det_model_name, det_model_backend = det_model_name.split('::')
     rec_model_name, rec_model_backend = rec_model_name.split('::')
-    return CnOcr(
+    model = CnOcr(
         det_model_name=det_model_name,
         det_model_backend=det_model_backend,
         rec_model_name=rec_model_name,
         rec_model_backend=rec_model_backend,
         det_more_configs=det_more_configs,
     )
+    if len(MODELS) > 50:
+        MODELS = {}
+    MODELS[(det_model_name, rec_model_name, config_str)] = model
+    return model
 
 
 def visualize_naive_result(img, det_model_name, std_out, box_score_thresh):
@@ -186,6 +199,66 @@ def main():
         '作者：<a href="https://www.breezedeus.com" target="_blank">Breezedeus</a> ，'
         '<a href="https://github.com/breezedeus" target="_blank">Github</a> 。</p>'
     )
+    example_func = functools.partial(
+        recognize,
+        # det_model_name='ch_PP-OCRv3_det::onnx',
+        rotated_bbox=True,
+        # use_angle_clf=False,
+        new_size=768,
+        box_score_thresh=0.3,
+        min_box_size=10,
+    )
+    examples = [
+        [
+            'ch_PP-OCRv3_det::onnx',
+            True,
+            'number-densenet_lite_136-fc::onnx',
+            False,
+            'docs/examples/card1-s.jpg',
+        ],
+        [
+            'ch_PP-OCRv3_det::onnx',
+            True,
+            'number-densenet_lite_136-fc::onnx',
+            False,
+            'docs/examples/card2-s.jpg',
+        ],
+        [
+            'ch_PP-OCRv3_det::onnx',
+            True,
+            'number-densenet_lite_136-fc::onnx',
+            False,
+            'docs/examples/cy1-s.jpg',
+        ],
+        [
+            'ch_PP-OCRv3_det::onnx',
+            False,
+            'densenet_lite_136-gru::onnx',
+            False,
+            'docs/examples/huochepiao.jpeg',
+        ],
+        [
+            'ch_PP-OCRv3_det::onnx',
+            False,
+            'densenet_lite_136-gru::onnx',
+            False,
+            'docs/examples/1_res.jpg',
+        ],
+        [
+            'db_shufflenet_v2::pytorch',
+            False,
+            'en_number_mobile_v2.0::onnx',
+            False,
+            'docs/examples/en_book1.jpeg',
+        ],
+        [
+            'db_shufflenet_v2::pytorch',
+            False,
+            'densenet_lite_136-gru::onnx',
+            True,
+            'docs/examples/beauty0.jpg',
+        ],
+    ]
 
     with gr.Blocks() as demo:
         gr.Markdown(
@@ -220,8 +293,8 @@ def main():
                 )
 
             with gr.Column(scale=3, variant='compact'):
-                gr.Markdown('### 选择待检测图片')
-                image_file = gr.Image(label='', type="pil", image_mode='RGB')
+                gr.Markdown('### 选择待识别图片')
+                image_file = gr.Image(label='待识别图片', type="pil", image_mode='RGB')
                 sub_btn = gr.Button("Submit", variant="primary")
                 out_image = gr.Image(label='识别结果', interactive=False, visible=False)
                 naive_warn = gr.Markdown(
@@ -246,6 +319,21 @@ def main():
                     outputs=[out_image, naive_warn, out_texts],
                 )
 
+        gr.Examples(
+            label='示例',
+            examples=examples,
+            inputs=[
+                det_model_name,
+                is_single_line,
+                rec_model_name,
+                use_angle_clf,
+                image_file,
+            ],
+            outputs=[out_image, naive_warn, out_texts],
+            fn=example_func,
+            cache_examples=os.getenv('CACHE_EXAMPLES') == '1',
+        )
+
     demo.queue(concurrency_count=4)
     demo.launch()