paddleOCR训练模型转化为推理模型教程

目录

将“检测”训练模型转化为推理模型

确定配置文件路径

修改配置文件

导出推理模型

将“识别”训练模型转化为推理模型

将“检测”训练模型转化为推理模型

确定配置文件路径

打开program.py脚本，找到class ArgsParser构造函数，修改为如下形式（注意相对路径填写正确）：

修改配置文件

在config/det文件夹下找到det_mv3_db_v1.1.yml文件，需要做如下两个修改：

• 
  将“checkpoints”指定为自己训练处的模型文件所在路径，如下图所示，指定的训练模型是300代的训练结果；
  
• 
  将“save_inference_dir”表示导出的推理模型路径，按照实际需要填写，如下图所示，推理模型导出路径为rec_InferenceModel文件夹下
  

（注意相对路径填写正确,“./ ”表示同级目录，“../”表示上级目录，“../../”表示上上级目录，以运行的.py文件为当前路径）

导出推理模型

运行脚本文件“export_model.py”，导出的文件共两个，分别是：“model”和“params”，如下图所示：

将“识别”训练模型转化为推理模型

方法同第1节，只需要修改如下项：

• 
  
  在“program.py”脚本中指定识别模型的配置文件；
  
  
• 
  
  修改识别模型的配置文件中“checkpoints”和“
  
  save_inference_dir
  
  ”
  
  

修改完成后，运行“export_model.py”脚本即可