Skip to content

Commit

Permalink
Merge pull request WenmuZhou#87 from WenmuZhou/master
Browse files Browse the repository at this point in the history
merge master
  • Loading branch information
WenmuZhou authored Jul 28, 2020
2 parents 74bb733 + 5be627b commit aa39734
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion doc/识别.md
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,7 @@ PytorchOCR提供了训练脚本和预测脚本,本节将以 CRNN 识别模型
1. `config.dataset.alphabet`: 字典文件
2. `config.dataset.train.dataset.file`: 训练集txt文件 or LMDB 数据集路径
3. `config.dataset.eval.dataset.file`: 验证集集txt文件 or LMDB 数据集路径
4. `config.model.head.n_class`: 分类字符数+背景

* 可选修改字段说明
1. `config.train_options.checkpoint_save_dir`: 模型和日志文件保存地址
Expand All @@ -64,4 +65,4 @@ PytorchOCR支持训练和评估交替进行, 可以在 `config.train_options`中
通过以下命令启动预测
```shell script
CUDA_VISIBLE_DEVICES=0 python3 tools/rec_infer.py --model_path '' --img_path ''
```
```

0 comments on commit aa39734

Please sign in to comment.