Files
rapidocr/python/rapid_structure/docs/README_Table.md
T
2022-12-19 14:07:19 +08:00

4.5 KiB

Rapid Table

PyPI

简介和说明
  • 该部分主要是做文档类图像的表格结构还原。该包依赖rapidocr_onnxruntime。
  • 具体来说,就是分析给定的表格图像,将表格还原为对应的html格式。
  • 目前支持两种类别的表格识别模型:中文和英文表格识别模型,具体可参见下面表格:
    模型类型 模型名称 模型大小
    英文 en_ppstructure_mobile_v2_SLANet.onnx 7.3M
    中文 ch_ppstructure_mobile_v2_SLANet.onnx 7.4M
  • 模型下载地址为:百度网盘 | Google Drive

使用方式

  1. pip安装
    • 由于模型较小,预先将英文表格识别模型(en_ppstructure_mobile_v2_SLANet.onnx)打包进了whl包内,如果做英文表格识别,可直接安装使用
      $ pip install rapid-table
      
  2. python脚本运行
     import cv2
     from rapid_table import RapidTable
    
     rapid_table = RapidTable()
    
     img = cv2.imread('test_images/table.jpg')
     table_html_str, _ = rapid_table(img)
     print(table_html_str)
    
  3. 终端运行
    # 假设在rapid_structure目录下
    $ rapid_table --img_path test_images/table.jpg
    
  4. 结果
    • 返回结果
      <html><body><table><thead><tr><td>Methods</td><td></td><td></td><td></td><td>FPS</td></tr></thead><tbody><tr><td>SegLink [26]</td><td>70.0</td><td>86d><td.0</td><td>77.0</td><td>8.9</td></tr><tr><td>PixelLink [4]</td><td>73.2</td><td>83.0</td><td>77.8</td><td></td></tr><tr><td>TextSnake [18]</td><td>73.9</td><td>83.2</td><td>78.3</td><td>1.1</td></tr><tr><td>TextField [37]</td><td>75.9</td><td>87.4</td><td>81.3</td><td>5.2</td></tr><tr><td>MSR[38]</td><td>76.7</td><td>87.87.4</td><td>81.7</td><td></td></tr><tr><td>FTSN [3]</td><td>77.1</td><td>87.6</td><td>82.0</td><td></td></tr><tr><td>LSE[30]</td><td>81.7</td><td>84.2</td><td>82.9</td><><ttd></td></tr><tr><td>CRAFT [2]</td><td>78.2</td><td>88.2</td><td>82.9</td><td>8.6</td></tr><tr><td>MCN[16]</td><td>79</td><td>88</td><td>83</td><td></td></tr><tr><td>ATRR</>[35]</td><td>82.1</td><td>85.2</td><td>83.6</td><td></td></tr><tr><td>PAN [34]</td><td>83.8</td><td>84.4</td><td>84.1</td><td>30.2</td></tr><tr><td>DB[12]</td><td>79.2</t91/d><td>91.5</td><td>84.9</td><td>32.0</td></tr><tr><td>DRRG[41]</td><td>82.30</td><td>88.05</td><td>85.08</td><td></td></tr><tr><td>Ours (SynText)</td><td>80.68</td><td>85<t..40</td><td>82.97</td><td>12.68</td></tr><tr><td>Ours (MLT-17)</td><td>84.54</td><td>86.62</td><td>85.57</td><td>12.31</td></tr></tbody></table></body></html>
      
    • 可视化结果
    <>
    MethodsFPS
    SegLink [26]70.086d>77.08.9
    PixelLink [4]73.283.077.8
    TextSnake [18]73.983.278.31.1
    TextField [37]75.987.481.35.2
    MSR[38]76.787.87.481.7
    FTSN [3]77.187.682.0
    LSE[30]81.784.282.9
    CRAFT [2]78.288.282.98.6
    MCN[16]798883
    ATRR[35]82.185.283.6
    PAN [34]83.884.484.130.2
    DB[12]79.2</t91/d>91.584.932.0
    DRRG[41]82.3088.0585.08
    Ours (SynText)80.688582.9712.68
    Ours (MLT-17)84.5486.6285.5712.31