使用rknn-toolkit2将paddleseg模型导出rknn模型

最新推荐文章于 2025-10-31 15:04:32 发布

原创

最新推荐文章于 2025-10-31 15:04:32 发布 · 1.3k 阅读

10 ·

CC 4.0 BY-SA版权

文章标签：

#rknn #paddleseg #onnx

本文介绍了如何在Python环境中安装paddle2onnx，将Paddle模型转换为ONNX格式，并进一步使用rknn-toolkits将模型转化为适配RKNN平台的模型，包括下载模型、执行预处理步骤和配置转换脚本。

安装paddle2onnx环境

首先创建一个python虚拟环境

conda create -n paddle2onnx python==3.10
source activate paddle2onnx

下载并安装 https://github.com/PaddlePaddle/Paddle2ONNX

cd Paddle2ONNX
pip install .

将paddle模型导出onnx模型

下载一个官方提供的模型，并解压

cd paddle2onnx/tools/paddle
wget https://bj.bcebos.com/paddlehub/fastdeploy/Portrait_PP_HumanSegV2_Lite_256x144_infer.tgz
tar -xvf Portrait_PP_HumanSegV2_Lite_256x144_infer.tgz

执行infer_shape

python paddle_infer_shape.py --model_dir portrait_pp_humansegv2_lite_256x144_inference_model \
                             --model_filename model.pdmodel \
                             --params_filename model.pd