CRAFT-Remade 项目使用教程

CRAFT-Remade 项目使用教程

CRAFT-RemadeImplementation of CRAFT Text Detection项目地址:https://gitcode.com/gh_mirrors/cr/CRAFT-Remade

1. 项目目录结构及介绍

CRAFT-Remade/
├── input/
├── model/
├── src/
├── train_synth/
├── train_weak_supervision/
├── .gitignore
├── LICENSE
├── README.md
├── config.py
├── environment.yml
├── main.py
└── requirements.txt

• input/: 存放自定义图像的文件夹。
• model/: 存放预训练模型的文件夹。
• src/: 存放项目源代码的文件夹。
• train_synth/: 用于合成训练数据的文件夹。
• train_weak_supervision/: 用于弱监督训练的文件夹。
• .gitignore: Git 忽略文件配置。
• LICENSE: 项目许可证文件。
• README.md: 项目介绍和使用说明。
• config.py: 项目配置文件。
• environment.yml: Conda 环境配置文件。
• main.py: 项目启动文件。
• requirements.txt: Python 依赖包列表。

2. 项目启动文件介绍

main.py 是 CRAFT-Remade 项目的启动文件。它包含了主要的命令行接口，用于运行文本检测任务。以下是启动文件的主要功能：

• synthesize: 用于在自定义图像上生成文本检测结果。
• train: 用于训练模型。
• evaluate: 用于评估模型性能。

使用示例：

python main.py synthesize --model=/model/final_model.pkl --folder=/input

3. 项目的配置文件介绍

config.py 是 CRAFT-Remade 项目的配置文件。它包含了项目运行所需的各种配置参数，如模型路径、数据路径、训练参数等。以下是配置文件的主要内容：

• MODEL_PATH: 预训练模型的路径。
• DATA_PATH: 训练数据的存放路径。
• TRAIN_PARAMS: 训练参数，如学习率、批量大小等。
• EVAL_PARAMS: 评估参数，如评估数据集路径等。

配置文件的使用示例：

# config.py
MODEL_PATH = "/model/final_model.pkl"
DATA_PATH = "/input"
TRAIN_PARAMS = {
    "learning_rate": 0.001,
    "batch_size": 32
}
EVAL_PARAMS = {
    "dataset_path": "/eval_data"
}

通过修改 config.py 文件中的参数，可以自定义项目的运行配置。

CRAFT-RemadeImplementation of CRAFT Text Detection项目地址:https://gitcode.com/gh_mirrors/cr/CRAFT-Remade