Spaces:
Running
Running
| # -*- encoding: utf-8 -*- | |
| # @Author: SWHL | |
| # @Contact: [email protected] | |
| import gradio as gr | |
| from PIL import Image | |
| img_path = "images/1.jpg" | |
| def test(): | |
| return Image.open(img_path) | |
| example_images = [ | |
| "images/1.jpg", | |
| "images/ch_en_num.jpg", | |
| "images/air_ticket.jpg", | |
| "images/car_plate.jpeg", | |
| "images/train_ticket.jpeg", | |
| "images/japan_2.jpg", | |
| "images/korean_1.jpg", | |
| ] | |
| custom_css = """ | |
| body {font-family: body {font-family: 'Helvetica Neue', Helvetica;} | |
| .gr-button {background-color: #4CAF50; color: white; border: none; padding: 10px 20px; border-radius: 5px;} | |
| .gr-button:hover {background-color: #45a049;} | |
| .gr-textbox {margin-bottom: 15px;} | |
| .example-button {background-color: #1E90FF; color: white; border: none; padding: 8px 15px; border-radius: 5px; margin: 5px;} | |
| .example-button:hover {background-color: #FF4500;} | |
| .tall-radio .gr-radio-item {padding: 15px 0; min-height: 50px; display: flex; align-items: center;} | |
| .tall-radio label {font-size: 16px;} | |
| .output-image, .input-image, .image-preview {height: 300px !important} | |
| """ | |
| with gr.Blocks( | |
| title="Rapid⚡OCR Demo", css="custom_css", theme=gr.themes.Soft() | |
| ) as demo: | |
| gr.Markdown( | |
| "<h1 style='text-align: center;'><a href='https://rapidai.github.io/RapidOCRDocs/' style='text-decoration: none;'>Rapid⚡OCR</a></h1>" | |
| ) | |
| gr.HTML( | |
| """ | |
| <div style="display: flex; justify-content: center; gap: 10px;"> | |
| <a href=""><img src="https://img.shields.io/badge/Python->=3.6-aff.svg"></a> | |
| <a href=""><img src="https://img.shields.io/badge/OS-Linux%2C%20Win%2C%20Mac-pink.svg"></a> | |
| <a href="https://pepy.tech/project/rapidocr"><img src="https://static.pepy.tech/personalized-badge/rapidocr?period=total&units=abbreviation&left_color=grey&right_color=blue&left_text=Downloads%20rapidocr"></a> | |
| <a href="https://pypi.org/project/rapidocr/"><img alt="PyPI" src="https://img.shields.io/pypi/v/rapidocr"></a> | |
| <a href="https://github.com/RapidAI/RapidOCR"><img src="https://img.shields.io/github/stars/RapidAI/RapidOCR?color=ccf"></a> | |
| </div> | |
| """ | |
| ) | |
| with gr.Row(): | |
| text_score = gr.Slider( | |
| label="text_score", | |
| minimum=0, | |
| maximum=1.0, | |
| value=0.5, | |
| step=0.1, | |
| info="文本识别结果是正确的置信度，值越大，显示出的识别结果更准确。存在漏检时，调低该值。取值范围：[0, 1.0]，默认值为0.5", | |
| ) | |
| box_thresh = gr.Slider( | |
| label="box_thresh", | |
| minimum=0, | |
| maximum=1.0, | |
| value=0.5, | |
| step=0.1, | |
| info="检测到的框是文本的概率，值越大，框中是文本的概率就越大。存在漏检时，调低该值。取值范围：[0, 1.0]，默认值为0.5", | |
| ) | |
| unclip_ratio = gr.Slider( | |
| label="unclip_ratio", | |
| minimum=1.5, | |
| maximum=2.0, | |
| value=1.6, | |
| step=0.1, | |
| info="控制文本检测框的大小，值越大，检测框整体越大。在出现框截断文字的情况，调大该值。取值范围：[1.5, 2.0]，默认值为1.6", | |
| ) | |
| img_input = gr.Image(label="Upload or Select Image", sources="upload") | |
| run_btn = gr.Button("Run") | |
| run_btn.click(test, inputs=img_input, outputs=gr.Image()) | |
| examples = gr.Examples( | |
| examples=example_images, | |
| examples_per_page=len(example_images), | |
| inputs=img_input, | |
| fn=lambda x: x, # 简单返回图片路径 | |
| outputs=img_input, | |
| cache_examples=False, | |
| ) | |
| if __name__ == "__main__": | |
| demo.launch(debug=True) | |