Chinese-ocr重新训练,ctpn + crnn
Chinese-ocr重新训练,ctpn + crnn项目代码参考:https://gitee.com/rothsword/CHINESE-OCRctpn训练注:见上一篇crnn训练1.预训练参数和路径匹配•keras版本 ./train/keras_train/train_batch.py model_path–指向预训练权重位置 MODEL_PATH—指向模型训练保存的位置ke...
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即可
更多推荐
所有评论(0)