Chinese-ocr重新训练,ctpn + crnn

项目代码参考:https://gitee.com/rothsword/CHINESE-OCR

ctpn训练

注:见上一篇

crnn训练

1.预训练参数和路径匹配

• keras版本 ./train/keras_train/train_batch.py model_path–指向预训练权重位置 MODEL_PATH—指向模型训练保存的位置
keras模型预训练权重:https://pan.baidu.com/s/1vTG6-i_bFMWxQ_7xF06usg

• pythorch版本./train/pytorch-train/crnn_main.py
parser.add_argument(
‘–crnn’,
help=“path to crnn (to continue training)”,
default=预训练权重的路径,看你下载的预训练权重在哪啦)
parser.add_argument(
‘–experiment’,
help=‘Where to store samples and models’,
default=模型训练的权重保存位置,这个自己指定)
pytorch预训练权重:https://pan.baidu.com/s/1LEDNHEr3luloB7eZK6GOeA

2.训练数据

运行create-dataset.sh文件生成训练数据

3.重新训练

(注意修改文件中的加载和保存模型的路径)
运行train-keras.sh或train-pytorch.sh

4.测试

项目/orc/model.py修改模型加载路径
modelPath = os.path.join(os.getcwd(), “模型保存路径”)

运行demo即可

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐