Chinese 您所在的位置:网站首页 me训练 Chinese

Chinese

2023-08-15 08:42| 来源: 网络整理| 查看: 265

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 实验室设备网 版权所有