Initial Commit

app.py

@@ -0,0 +1,34 @@
+import gradio as gr
+
+from pp_ocr import inference_img, inference_json
+
+title = "基于PP-OCRv3文本识别"
+description = """
+    PaddleOCR是百度开源的超轻量级OCR模型库，提供了数十种文本检测、识别模型，旨在打造一套丰富、领先、实用的文字检测、识别模型/工具库。
+    > 项目地址：PaddleOCR github 地址： https://github.com/PaddlePaddle/PaddleOCR
+"""
+
+with gr.Blocks() as app:
+    gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>"
+                + title
+                + "</h1>")
+    gr.Markdown(description)
+    with gr.Tab("图片"):
+        with gr.Row():
+            with gr.Column():
+                img_input = gr.Image()
+                img_btn = gr.Button("识别")
+            with gr.Column():
+                img_output = gr.Image(label="Result")
+    with gr.Tab("JSON"):
+        with gr.Row():
+            with gr.Column():
+                json_input = gr.Image()
+                json_btn = gr.Button("识别")
+            with gr.Column():
+                json_output = gr.Json(label="Result")
+
+    img_btn.click(inference_img, inputs=img_input, outputs=img_output)
+    json_btn.click(inference_json, inputs=json_input, outputs=json_output)
+
+app.launch()

pp_ocr.py

@@ -0,0 +1,18 @@
+import tempfile
+import os
+
+import paddlehub as hub
+from PIL import Image
+
+pp_ocrv3 = hub.Module(name="ch_pp-ocrv3")
+
+def inference_img(img):
+    with tempfile.TemporaryDirectory() as tempdir_name:
+        pp_ocrv3.recognize_text(images=[img], use_gpu=False, output_dir=tempdir_name, visualization=True)
+        result_names = os.listdir(tempdir_name)
+        result_image = Image.open(os.path.join(tempdir_name, result_names[0]))
+        return result_image
+    
+def inference_json(img):
+        results = pp_ocrv3.recognize_text(images=[img], use_gpu=False, visualization=False)
+        return results

requirements.txt

@@ -0,0 +1,3 @@
+paddlepaddle==2.4.2
+paddlehub
+gradio