pdmodel从动态模型转成静态onnx

1.下载项目

git clone https://github.com/jiangjiajun/PaddleUtils.git

2.新建两个新的文件夹

第一个文件夹放两个必要文件

第二个文件夹可以设置为空,用来存放转换后的模型

如图:

3.在终端运行

 python paddle/paddle_infer_shape.py --model_dir ch_PP-OCRv2_det_infer/ --model_filename inference.pdmodel --params_filename inference.pdiparams --save_dir new_model --input_shape_dict="{'x':[1,3,960,960]}" 

通常需要修改第0个参数,第2、3个参数;分别为batch_size/weight/height

4.转化静态onnx

安装paddle2onnx

 paddle2onnx --model_dir new_model/            --model_filename inference.pdmodel            --params_filename inference.pdiparams  --save_file cht3.onnx

5.可视化

之前

之后

最后的onnx

6.大功告成!

相关推荐

  1. 【深度学习】Pytorch模型Onnx

    2024-02-03 13:50:02       39 阅读
  2. tensorflow | onnx模型pb

    2024-02-03 13:50:02       45 阅读
  3. 模型部署之——ONNX模型RKNN

    2024-02-03 13:50:02       52 阅读
  4. bert pytorch模型onnx,并改变输入输出

    2024-02-03 13:50:02       45 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-02-03 13:50:02       91 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-02-03 13:50:02       97 阅读
  3. 在Django里面运行非项目文件

    2024-02-03 13:50:02       78 阅读
  4. Python语言-面向对象

    2024-02-03 13:50:02       88 阅读

热门阅读

  1. 鸿蒙设置沉浸式状态栏、全屏、获取导航栏高度

    2024-02-03 13:50:02       203 阅读
  2. json文件缺少的语言key的检测

    2024-02-03 13:50:02       52 阅读
  3. 【Python】requests库的介绍及用法

    2024-02-03 13:50:02       42 阅读
  4. python--第三方包的使用

    2024-02-03 13:50:02       48 阅读
  5. Vue3 vant4 底部tabbar图标如何修改为自定义图片

    2024-02-03 13:50:02       51 阅读
  6. ChatGPT等大模型AI能干什么?

    2024-02-03 13:50:02       45 阅读
  7. 力扣0120——三角形最小路径和

    2024-02-03 13:50:02       55 阅读