PaddlePaddle推理模型利用Paddle2ONNX转换成onnx模型

ops/2024/12/20 13:38:35/

训练paddleocr模型,可以按照如下教程或者直接百度其他的

PaddleOCR训练自己的数据集(中英文)(全套)和C#部署(半套)_c# paddleocr-CSDN博客

将训练后的模型转换成推理模型

-c后面是你配置的det  的yaml文件   -o Global.pretrained_model=后面是你训练后的模型 latest

lobal.save_inference_dir=保存转换后的推理模型位置

python tools/export_model.py -c "./configs/endet_enrec_encls/en_PP_OCRV3_det_cml.yml" -o Global.pretrained_model="./output/det/ch_PP-OCR_v3_det3/latest" Global.save_inference_dir="./models/inf_det/"

 rec的转换和上方的det一样

 转换后是如下

 然后在命令行中pip下载paddle2onnx和onnxruntime

# 安装paddle2onnx
pip install paddle2onnx -i https://pypi.tuna.tsinghua.edu.cn/simple
# 安装 ONNXRuntime,建议安装 1.9.0 版本,可根据环境更换版本号
pip install onnxruntime==1.9.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

 转换命令如下

paddle2onnx --model_dir saved_inference_model \--model_filename model.pdmodel \--params_filename model.pdiparams \--save_file model.onnx \--enable_dev_version True

 

 如果看不明白,下方是我的转换命令

model_dir 是你训练模型的文件夹

model_filename 是结尾是 .pdmodel文件

params_filename 是结尾是 .pdiparams文件

save_file 保存转换后的onnx模型

当然rec和det以及旋转模型也是如下,一共有三个。

paddle2onnx --model_dir H:\\DL\\PaddleOCR-main\\models\\ch_PP-OCRv3_rec_infer--model_filename inference.pdmodel --params_filename inference.pdiparams --save_file F:\\Desk\\159\\inference.onnx

测试推理 

onnxruntime推理

python tools/infer/predict_system.py --use_gpu=False --use_onnx=True \
--det_model_dir=./inference/det_onnx/model.onnx  \
--rec_model_dir=./inference/rec_onnx/model.onnx  \
--cls_model_dir=./inference/cls_onnx/model.onnx  \
--image_dir=./deploy/lite/imgs/lite_demo.png

 优化模型命令

python -m paddle2onnx.optimize --input_model model.onnx --output_model optimized_model.onnx


http://www.ppmy.cn/ops/143504.html

相关文章

1157:哥德巴赫猜想

为了弥补我过往的错误&#xff01;我决定上吊&#xff01;啊不不不&#xff0c;是割脉&#xff01;啊说错了 再更一期&#xff01;请原谅我&#xff01; 来一个函数吧&#xff01; 1157&#xff1a;哥德巴赫猜想 #include<iostream> #include<cstdio> #include&…

MES系统工作流的单元测试方案

MES系统工作流的单元测试方案 在基于Java实现的MES系统中&#xff0c;若算子组成工作流并通过JSON传递数据&#xff0c;后端解析JSON后执行业务逻辑的流程&#xff0c;单元测试的核心是确保以下内容的正确性&#xff1a; 算子功能的正确性&#xff08;每个算子单独的逻辑&…

CSS边框的样式

边框阴影 让元素更有立体感 img {box-shadow: 2px 10px 5px 20px #ff0000;border-radius: 44px;}语法&#xff1a;box-shadow&#xff1a;值1 值2 值3 值4 值5 值1&#xff1a;水平阴影的位置值2&#xff1a;垂直阴影的位置值3&#xff1a;模糊距离值4&#xff1a;阴影的尺寸…

JMeter配置原件-计数器

一、面临的问题&#xff1a; 由于本人的【函数助手对话框】中counter计数器每次加2&#xff0c;且只显示偶数(如下图所示)&#xff0c;因此借助【配置原件-计数器】来实现计数功能。 如果有大佬知道解决方式&#xff0c;麻烦评论区解答一下&#xff0c;谢谢。 二、配置原件-c…

各向同性重建(3D荧光图像)

文章目录 一、基本知识1.1、各向同性&#xff08;isotropic&#xff09; 各向异性&#xff08;anisotropic&#xff09;1.2、像素尺寸 像素间距 像素分辨率1.3、点扩散函数&#xff08;PSF&#xff09;1.3.1、高斯函数 二维高斯PSF1.3.2、二维高斯PSF&#xff1a;代码生成 …

Spring IOC 和 AOP的学习笔记

Spring框架是java开发行业的标准 Spring全家桶 Web&#xff1a;Spring Web MVC/Spring MVC、Spring Web Flux 持久层&#xff1a;Spring Data / Spring Data JPA 、Spring Data Redis 、Spring Data MongoDB 安全校验&#xff1a;Spring Security 构建工程脚手架&#xff…

如何实现《黑神话:悟空》人物换脸?ToDesk云电脑AIGC版实操教程

《黑神话&#xff1a;悟空》作为2024年下半年颇为流行的一款游戏&#xff0c;热度很高&#xff0c;广受国内外玩家欢迎。作为首款国产3A游戏&#xff0c;它不仅在电竞爱好者圈子中受到追捧&#xff0c;亦是可以说是在年轻群体中基本到了无人不知、无人不晓的程度&#xff01; …

folly库高性能futex源码探究

Futex(快速用户空间互斥量)是一个32位的无符号整型原子变量,它允许对该值执行futex()系统调用。它以模板化的方式设计,使其能够与确定性调度测试(DeterministicSchedule)正确交互。 h源码: enum class FutexResult {VALUE_CHANGED, /* futex value didnt match expect…