读光-表格结构识别-有线表格使用指南

最新推荐文章于 2024-08-27 14:10:44 发布

kjzd123

最新推荐文章于 2024-08-27 14:10:44 发布

阅读量358

点赞数 6

文章标签： pdf 人工智能 python

本文链接：https://blog.csdn.net/kjzd123/article/details/138523742

版权

1.下载代码安装环境

git clone https://github.com/RapidAI/TableStructureRec.git

pip install -r requirements.txt

2.修改图片存放路径

在这里插入图片描述

运行main代码得到多个html文件使用以下代码将多个html文件合成一个html文件

from bs4 import BeautifulSoup
import os

# 设置输入文件夹和输出文件路径
input_folder = r'input_images'
output_file = r'/merged.html'

# 存储所有表格内容的列表
all_tables = []

# 遍历文件夹中的所有HTML文件
for filename in os.listdir(input_folder):
    if filename.endswith('.html'):
        filepath = os.path.join(input_folder, filename)

        # 读取HTML文件内容
        with open(filepath, 'r', encoding='utf-8') as file:
            html_content = file.read()

        # 使用Beautiful Soup解析HTML
        soup = BeautifulSoup(html_content, 'html.parser')

        # 查找所有表格
        tables = soup.find_all('table')

        # 将每个表格的内容添加到列表中
        for table in tables:
            all_tables.append(str(table))  # 将Beautiful Soup对象转换为字符串存储

# 创建输出HTML文件
with open(output_file, 'w', encoding='utf-8') as outfile:
    # 写入HTML头部和开始标签
    outfile.write('<!DOCTYPE html>\n<html>\n<head>\n<title>Merged Tables</title>\n</head>\n<body>\n')

    # 逐一将所有表格内容写入输出文件
    for table_html in all_tables:
        outfile.write(table_html)
        outfile.write('\n')  # 每个表格之间添加换行

    # 写入HTML结束标签
    outfile.write('</body>\n</html>\n')

print(f'Merged HTML file saved to: {output_file}')