Spaces:

SWHL
/

RapidStructureDemo

Running

App Files Files Community

SWHL commited on Jun 15, 2023

Commit

d3a7100

•

1 Parent(s): c9ad304

Update files

Browse files

Files changed (5) hide show

app.py +84 -10
pages/1_rapid_orientation.py +0 -46
pages/2_rapid_layout.py +0 -35
pages/3_rapid_table.py +0 -35
requirements.txt +2 -1

app.py CHANGED Viewed

@@ -1,9 +1,44 @@
 # -*- encoding: utf-8 -*-
 # @Author: SWHL
 # @Contact: liekkaskono@163.com
 import numpy as np
 import streamlit as st
 from PIL import Image
 if __name__ == '__main__':
@@ -20,16 +55,55 @@ if __name__ == '__main__':
     img_suffix = ["png", "jpg", "jpeg"]
-    st.markdown('##### rapid-orientation (文档图像分类) demo')
-    img_empty = st.empty()
-    img_file_buffer = st.file_uploader("Upload an image",
-                                       type=img_suffix,
-                                       label_visibility='hidden')
     if img_file_buffer:
         image = Image.open(img_file_buffer)
-        img_array = np.array(image)
-        if image:
-            img_empty.image(image, use_column_width=True)
-    st.markdown('##### rapid-layout (版面分析) demo')
-    st.markdown('##### rapid-table (表格还原) demo')

 # -*- encoding: utf-8 -*-
 # @Author: SWHL
 # @Contact: liekkaskono@163.com
+import copy
+import cv2
 import numpy as np
 import streamlit as st
 from PIL import Image
+from rapid_layout import RapidLayout
+from rapid_orientation import RapidOrientation
+from rapid_table import RapidTable
+orientation_engine = RapidOrientation()
+layout_engine = RapidLayout()
+table_engine = RapidTable()
+def vis_layout(img: np.ndarray, layout_res: list) -> None:
+    tmp_img = copy.deepcopy(img)
+    for v in layout_res:
+        bbox = np.round(v['bbox']).astype(np.int32)
+        label = v['label']
+        start_point = (bbox[0], bbox[1])
+        end_point = (bbox[2], bbox[3])
+        cv2.rectangle(tmp_img, start_point, end_point, (0, 255, 0), 2)
+        cv2.putText(tmp_img, label, start_point,
+                    cv2.FONT_HERSHEY_COMPLEX, 1, (0, 0, 255), 2)
+    return tmp_img
+def vis_table(table_res) -> str:
+    style_res = '''<style>td {border-left: 1px solid;border-bottom:1px solid;}
+                   table, th {border-top:1px solid;font-size: 10px;
+                   border-collapse: collapse;border-right: 1px solid;}
+                </style>'''
+    prefix_table, suffix_table = table_res.split('<body>')
+    new_table_res = f'{prefix_table}{style_res}<body>{suffix_table}'
+    return new_table_res
 if __name__ == '__main__':
     img_suffix = ["png", "jpg", "jpeg"]
+    st.markdown('##### 文档图像方向分类')
+    img_file_buffer = st.file_uploader("Upload an image", type=img_suffix,
+                                       key='orientation',
+                                       label_visibility='collapsed')
+    col1, col2 = st.columns([5, 5])
+    img_empty = col1.empty()
+    if img_file_buffer:
+        image = Image.open(img_file_buffer)
+        img = np.array(image)
+        img_empty.image(image, use_column_width=True)
+        img_array = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
+        orientation_res, elapse = orientation_engine(img)
+        col2.markdown(f'- 方向分类结果：{orientation_res}° \n - 耗费时间：{elapse:.4f}s')
+    st.markdown('##### 文档图像版面分析')
+    img_file_buffer = st.file_uploader("Upload an image", type=img_suffix,
+                                       key='layout',
+                                       label_visibility='collapsed')
+    layout_col1, layout_col2 = st.columns([5, 5])
+    img_empty = layout_col1.empty()
     if img_file_buffer:
         image = Image.open(img_file_buffer)
+        img = np.array(image)
+        img_empty.image(image, use_column_width=True)
+        img_array = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
+        layout_res, _ = layout_engine(img)
+        drawed_img = vis_layout(img, layout_res)
+        layout_col2.image(drawed_img, use_column_width=True)
+    st.markdown('##### 表格还原')
+    img_file_buffer = st.file_uploader("Upload an image", type=img_suffix,
+                                       key='table',
+                                       label_visibility='collapsed')
+    table_col1, table_col2 = st.columns([5, 5])
+    img_empty = table_col1.empty()
+    if img_file_buffer:
+        image = Image.open(img_file_buffer)
+        img = np.array(image)
+        img_empty.image(image, use_column_width=True)
+        img_array = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
+        table_html_str, _ = table_engine(img)
+        table_html_str = vis_table(table_html_str)
+        table_col2.markdown(table_html_str, unsafe_allow_html=True)

pages/1_rapid_orientation.py DELETED Viewed

@@ -1,46 +0,0 @@
-# -*- encoding: utf-8 -*-
-# @Author: SWHL
-# @Contact: liekkaskono@163.com
-import cv2
-import numpy as np
-import streamlit as st
-from PIL import Image
-from rapid_orientation import RapidOrientation
-orientation_engine = RapidOrientation()
-if __name__ == '__main__':
-    st.markdown("<h1 style='text-align: center;'><a href='https://github.com/RapidAI/RapidStructure/blob/main/docs/README_Orientation.md' style='text-decoration: none'>rapid-orientation</a></h1>", unsafe_allow_html=True)
-    st.markdown("""
-    <p align="left">
-        <a href=""><img src="https://img.shields.io/badge/Python->=3.7,<=3.10-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/rapid-orientation"><img src="https://static.pepy.tech/personalized-badge/rapid-orientation?period=total&units=abbreviation&left_color=grey&right_color=blue&left_text=rapid-orientation"></a>
-    </p>
-    #### 简介
-    - 该部分主要是做含文字图像方向分类模型。模型来源：[PaddleClas 含文字图像方向分类模型](https://github.com/PaddlePaddle/PaddleClas/blob/177e4be74639c0960efeae2c5166d3226c9a02eb/docs/zh_CN/models/PULC/PULC_text_image_orientation.md)
-        | 模型类型  |        模型名称         | 模型大小 |                           支持种类                           |
-        |:---:|:---:|:---:|:---:|
-        |   四方向分类   |   `rapid_orientation.onnx`   |  6.5M | `0 90 180 270`|
-    #### 试一下
-    """, unsafe_allow_html=True)
-    img_suffix = ["png", "jpg", "jpeg"]
-    img_file_buffer = st.file_uploader("Upload an image", type=img_suffix)
-    col1, col2 = st.columns([5, 5])
-    img_empty = col1.empty()
-    if img_file_buffer:
-        image = Image.open(img_file_buffer)
-        img = np.array(image)
-        img_empty.image(image, use_column_width=True)
-        img_array = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
-        orientation_res, elapse = orientation_engine(img)
-        col2.markdown(f'- 方向分类结果：{orientation_res}° \n - 耗费时间：{elapse:.4f}s')

pages/2_rapid_layout.py DELETED Viewed

@@ -1,35 +0,0 @@
-# -*- encoding: utf-8 -*-
-# @Author: SWHL
-# @Contact: liekkaskono@163.com
-import numpy as np
-import streamlit as st
-from PIL import Image
-if __name__ == '__main__':
-    st.markdown("<h1 style='text-align: center;'><a href='https://github.com/RapidAI/RapidStructure' style='text-decoration: none'>Rapid Structure</a></h1>", unsafe_allow_html=True)
-    st.markdown("""
-    <p align="left">
-        <a href=""><img src="https://img.shields.io/badge/Python->=3.7,<=3.10-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/rapid-layout"><img src="https://static.pepy.tech/personalized-badge/rapid-layout?period=total&units=abbreviation&left_color=grey&right_color=blue&left_text=rapid-layout"></a>
-        <a href="https://pepy.tech/project/rapid-orientation"><img src="https://static.pepy.tech/personalized-badge/rapid-orientation?period=total&units=abbreviation&left_color=grey&right_color=blue&left_text=rapid-orientation"></a>
-        <a href="https://pepy.tech/project/rapid-table"><img src="https://static.pepy.tech/personalized-badge/rapid-table?period=total&units=abbreviation&left_color=grey&right_color=blue&left_text=rapid-table"></a>
-    </p>
-    """, unsafe_allow_html=True)
-    img_suffix = ["png", "jpg", "jpeg"]
-    st.markdown('##### rapid-orientation (文档图像分类) demo')
-    img_empty = st.empty()
-    img_file_buffer = st.file_uploader("Upload an image",
-                                       type=img_suffix,
-                                       label_visibility='hidden')
-    if img_file_buffer:
-        image = Image.open(img_file_buffer)
-        img_array = np.array(image)
-        if image:
-            img_empty.image(image, use_column_width=True)
-    st.markdown('##### rapid-layout (版面分析) demo')
-    st.markdown('##### rapid-table (表格还原) demo')

pages/3_rapid_table.py DELETED Viewed

@@ -1,35 +0,0 @@
-# -*- encoding: utf-8 -*-
-# @Author: SWHL
-# @Contact: liekkaskono@163.com
-import numpy as np
-import streamlit as st
-from PIL import Image
-if __name__ == '__main__':
-    st.markdown("<h1 style='text-align: center;'><a href='https://github.com/RapidAI/RapidStructure' style='text-decoration: none'>Rapid Structure</a></h1>", unsafe_allow_html=True)
-    st.markdown("""
-    <p align="left">
-        <a href=""><img src="https://img.shields.io/badge/Python->=3.7,<=3.10-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/rapid-layout"><img src="https://static.pepy.tech/personalized-badge/rapid-layout?period=total&units=abbreviation&left_color=grey&right_color=blue&left_text=rapid-layout"></a>
-        <a href="https://pepy.tech/project/rapid-orientation"><img src="https://static.pepy.tech/personalized-badge/rapid-orientation?period=total&units=abbreviation&left_color=grey&right_color=blue&left_text=rapid-orientation"></a>
-        <a href="https://pepy.tech/project/rapid-table"><img src="https://static.pepy.tech/personalized-badge/rapid-table?period=total&units=abbreviation&left_color=grey&right_color=blue&left_text=rapid-table"></a>
-    </p>
-    """, unsafe_allow_html=True)
-    img_suffix = ["png", "jpg", "jpeg"]
-    st.markdown('##### rapid-orientation (文档图像分类) demo')
-    img_empty = st.empty()
-    img_file_buffer = st.file_uploader("Upload an image",
-                                       type=img_suffix,
-                                       label_visibility='hidden')
-    if img_file_buffer:
-        image = Image.open(img_file_buffer)
-        img_array = np.array(image)
-        if image:
-            img_empty.image(image, use_column_width=True)
-    st.markdown('##### rapid-layout (版面分析) demo')
-    st.markdown('##### rapid-table (表格还原) demo')

requirements.txt CHANGED Viewed

@@ -3,4 +3,5 @@ rapid-layout
 rapid-table
 rapid-orientation
 Pillow
-numpy

 rapid-table
 rapid-orientation
 Pillow
+numpy
+opencv-python