【亲测免费】 360LayoutAnalysis 项目使用指南

360LayoutAnalysis 项目使用指南

1. 项目的目录结构及介绍

360LayoutAnalysis/
├── LICENSE.txt
├── README.md
├── README_EN.md
├── requirements.txt
├── 360LayoutAnalysis开源模型许可证.txt
├── main.py
├── config/
│   ├── default_config.yaml
│   └── custom_config.yaml
├── models/
│   ├── model_weights/
│   │   └── 360LayoutAnalysis.pth
│   └── utils/
│       └── model_utils.py
├── data/
│   ├── sample_data/
│   │   └── sample.pdf
│   └── processed_data/
│       └── processed_sample.json
└── tests/
    └── test_model.py

• LICENSE.txt: 项目许可证文件。
• README.md: 项目介绍和使用说明。
• README_EN.md: 英文版本的项目介绍和使用说明。
• requirements.txt: 项目依赖文件。
• 360LayoutAnalysis开源模型许可证.txt: 模型许可证文件。
• main.py: 项目启动文件。
• config/: 配置文件目录。
  • default_config.yaml: 默认配置文件。
  • custom_config.yaml: 自定义配置文件。
• models/: 模型相关文件目录。
  • model_weights/: 模型权重文件目录。
    • 360LayoutAnalysis.pth: 模型权重文件。
  • utils/: 模型工具文件目录。
    • model_utils.py: 模型工具文件。
• data/: 数据文件目录。
  • sample_data/: 示例数据文件目录。
    • sample.pdf: 示例PDF文件。
  • processed_data/: 处理后的数据文件目录。
    • processed_sample.json: 处理后的示例数据文件。
• tests/: 测试文件目录。
  • test_model.py: 模型测试文件。

2. 项目的启动文件介绍

main.py 是项目的启动文件，负责加载配置、初始化模型和启动文档分析任务。以下是 main.py 的主要功能：

import argparse
import yaml
from models.utils.model_utils import load_model
from config.default_config import DEFAULT_CONFIG

def main():
    parser = argparse.ArgumentParser(description="360LayoutAnalysis 文档分析工具")
    parser.add_argument("--config", type=str, default="config/default_config.yaml", help="配置文件路径")
    args = parser.parse_args()

    with open(args.config, 'r') as f:
        config = yaml.safe_load(f)

    model = load_model(config)
    # 启动文档分析任务
    # ...

if __name__ == "__main__":
    main()

3. 项目的配置文件介绍

config/ 目录下包含两个配置文件：

• default_config.yaml: 默认配置文件，包含模型路径、数据路径等默认设置。
• custom_config.yaml: 自定义配置文件，用户可以根据需要修改配置。

以下是 default_config.yaml 的部分内容示例：

model_path: "models/model_weights/360LayoutAnalysis.pth"
data_path: "data/sample_data/sample.pdf"
output_path: "data/processed_data/processed_sample.json"

用户可以根据实际需求修改这些配置项，以适应不同的文档分析任务。