PaddleOCR文字识别使用教程(训练自己的数据集)(一)

目录

    • 项目目录结构
    • 数据集准备
        • label
        • img
    • 参数修改
    • 训练
    • 预测

项目目录结构

PaddleOCR文字识别使用教程(训练自己的数据集)(一)_第1张图片
链接:https://pan.baidu.com/s/1wbZvvonRBL3xy7xO3w57eA
提取码:8888

开源项目链接:https://github.com/PaddlePaddle/PaddleOCR/blob/release/2.3/doc/doc_en/recognition_en.md

数据集准备

label

        路径         识别的内容

PaddleOCR文字识别使用教程(训练自己的数据集)(一)_第2张图片

img

PaddleOCR文字识别使用教程(训练自己的数据集)(一)_第3张图片
当然你的图片名称要和你label里面的内容 一 一 对应。

参数修改

配置文件所在位置:
PaddleOCR文字识别使用教程(训练自己的数据集)(一)_第4张图片
PaddleOCR文字识别使用教程(训练自己的数据集)(一)_第5张图片

PaddleOCR文字识别使用教程(训练自己的数据集)(一)_第6张图片

训练

如果前面的内容都没有问题的话
在终端输入

python tools/train.py -c configs/rec/rec_icdar15_train.yml

文字识别板块使用GPU很快速的!
PaddleOCR文字识别使用教程(训练自己的数据集)(一)_第7张图片

预测

python tools/infer_rec.py -c configs/rec/rec_icdar15_train.yml -o Global.checkpoints=./output/rec_CRNN/best_accuracy  Global.infer_img=/home/song/wk_test/home/aistudio/PaddleOCR/iimg/213_3.jpg

PaddleOCR文字识别使用教程(训练自己的数据集)(一)_第8张图片

checkpoints参数在这里是你加载的三个配置权重模型,paddle训练完有三个文件,这里加载的时最好的模型(paddle会自动根据你所设置的文件名称对应加载相应的模型)

infer_img 参数在这里是你要预测的图片

希望这篇文章对你有用!
谢谢点赞评论!

你可能感兴趣的:(深度学习,paddlepaddle,卷积,深度可分离卷积,ocr,paddleocr,文字识别)