3_paddleOCR将训练模型转化为推理模型教程

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”脚本即可