DPText-DETR 项目使用教程

DPText-DETR 项目使用教程

DPText-DETR [AAAI 2023 Oral] DPText-DETR: Towards Better Scene Text Detection with Dynamic Points in Transformer 项目地址: https://gitcode.com/gh_mirrors/dp/DPText-DETR

1. 项目目录结构及介绍

DPText-DETR/
├── configs/
│   └── DPText_DETR/
│       ├── Pretrain/
│       │   └── R_50_poly.yaml
│       └── TotalText/
│           └── R_50_poly.yaml
├── demo/
│   └── demo.py
├── figs/
├── tools/
│   └── train_net.py
├── LICENSE
├── README.md
├── process_positional_label.py
├── setup.py
└── ...

目录结构介绍

• configs/: 存放项目的配置文件，包括预训练和微调的配置文件。
  • DPText_DETR/: 配置文件的主要目录。
    • Pretrain/: 预训练配置文件。
    • TotalText/: 针对 TotalText 数据集的微调配置文件。
• demo/: 存放演示脚本，用于图像的推理和可视化。
• figs/: 存放项目相关的图片文件。
• tools/: 存放训练和评估的脚本。
• LICENSE: 项目的许可证文件。
• README.md: 项目的介绍文档。
• process_positional_label.py: 处理位置标签的脚本。
• setup.py: 项目的安装脚本。

2. 项目启动文件介绍

tools/train_net.py

这是项目的主要启动文件，用于训练和评估模型。

主要功能

• 训练模型: 使用配置文件启动训练过程。
• 评估模型: 加载预训练模型进行评估。

使用示例

# 训练模型
python tools/train_net.py --config-file configs/DPText_DETR/Pretrain/R_50_poly.yaml --num-gpus 4

# 评估模型
python tools/train_net.py --config-file configs/DPText_DETR/TotalText/R_50_poly.yaml --eval-only MODEL.WEIGHTS path/to/model.pth

3. 项目的配置文件介绍

configs/DPText_DETR/Pretrain/R_50_poly.yaml

这是用于预训练模型的配置文件。

主要配置项

• DATASETS: 指定训练和测试数据集。
• SOLVER: 定义优化器和学习率等参数。
• MODEL: 定义模型的结构和参数。

configs/DPText_DETR/TotalText/R_50_poly.yaml

这是用于微调模型的配置文件，针对 TotalText 数据集。

主要配置项

• DATASETS: 指定训练和测试数据集。
• SOLVER: 定义优化器和学习率等参数。
• MODEL: 定义模型的结构和参数。

通过这些配置文件，用户可以根据需要调整模型的训练和评估参数。

DPText-DETR [AAAI 2023 Oral] DPText-DETR: Towards Better Scene Text Detection with Dynamic Points in Transformer 项目地址: https://gitcode.com/gh_mirrors/dp/DPText-DETR