Chinese | 您所在的位置:网站首页 › me训练 › Chinese |
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即可 |
CopyRight 2018-2019 实验室设备网 版权所有 |