当前位置:   article > 正文

PaddleOCRV4训练自己的模型(4)------模型推理及导出_paddleocr ocrv4 训练

paddleocr ocrv4 训练

一、Det模型推理:

(1)上一篇文章只讲了推理的实现方法,没有展示结果,这里顺带展示一下结果。

         因为训练定位模型的时候是整图训练,所以推理的时候也是整图推理。

(2)在推理的时候可以通过修改这个值来定义推理结果的保存路径

二、Rec模型推理

(1)在训练Rec的时候是只对文字部分进行训练的,推理的时候如果输入整图,推理时间可能会增加,效果可能也会下降(没测试过)  infer_rec.py的输出结果是一个TXT文件,一行对应每一张图的输出结果

三、模型导出

(1)在实际工作中,推理的时候定位和识别模型一般都是同时进行的。图片先经过定位模型进行检测,然后将定位出文字的区域送进识别模型进行识别。所以我们需要将模型导出,使得这两个模型可以串联处理。

(2)det导出参数:打开export_model.py文件,然后配置运行参数。yml文件对应自己训练的yml

 

  1. -c
  2. ../configs/det/ch_PP-OCRv4/ch_PP-OCRv4_det_cml.yml
  3. -o
  4. Global.pretrained_model=../output/ch_PP-OCRv4/best_accuracy.pdparams
  5. Global.save_inference_dir=../inference_model/det/

(3)rec导出参数:打开export_model.py文件,然后配置运行参数。

  1. -c
  2. ../configs/rec/PP-OCRv4/ch_PP-OCRv4_rec.yml
  3. -o
  4. Global.pretrained_model=..output/rec_ppocr_v4/best_accuracy.pdparams
  5. Global.save_inference_dir=../inference_model/rec/

四、最终推理

(1)打开predict_system.py文件

(2)配置运行参数,这个py文件在代码里面的rec_char_dict_path参数默认是./ppocr/utils/ppocr_keys_v1.txt,用cmd窗口跑的话可能没问题,但是用pycharm来跑的话会报错找不到这个文件。这是相对路径的原因,所以在参数里面修改rec_char_dict_path。

  1. --image_dir=../../train_data/det/test
  2. --det_model_dir=../../inference_model/det/Student
  3. --rec_model_dir=../../inference_model/rec
  4. --rec_char_dict_path=../../ppocr/utils/ppocr_keys_v1.txt

配置完运行参数就可以运行代码了,输出结果如下(左边是定位的结果,右边是文字识别的结果,识别模型训练的效果不好):

五、小技巧

由于使用pycharm运行,所以经常需要修改对应py文件的配置参数。所以可以把配置参数copy下来,这样就不用经常找路径了。这也是使用pycharm运行比较麻烦的一点。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小蓝xlanll/article/detail/679067
推荐阅读
相关标签
  

闽ICP备14008679号