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
文件中的参数,可以自定义项目的运行配置。