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

目录

安装paddle2onnx环境

将paddle模型导出onnx模型

安装rknn-toolkits

转化rknn模型


安装paddle2onnx环境

将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.pdiparams
    --save_dir portrait_pp_humansegv2_lite_256x144_inference_model
    --input_shape_dict="{'x':[1,3,144,255]}"

  • 导出onnx模型

    paddle2onnx --model_dir portrait_pp_humansegv2_lite_256x144_inference_model
    --model_filename model.pdmodel
    --params_filename model.pdiparams
    --save_file portrait_pp_humansegv2_lite_256x144_inference_model/portrait_pp_humansegv2_lite_256x144_inference_model.onnx
    --enable_dev_version True

安装rknn-toolkits

转化rknn模型

  • 我们直接使用FastDeploy中的python脚本,地址在tools/rknpu2/export.py

  • 编写一个yaml配置文件myseg.yaml,文件中的onnx模型是刚才转换得到的,修改成自己的目录,dataset.txt 在FastDeploy中有,拷贝到模型目录中,指定输出目录,配置完成后执行

    mean:
    -
    - 127.5
    - 127.5
    - 127.5
    std:
    -
    - 127.5
    - 127.5
    - 127.5
    model_path: ./Portrait_PP_HumanSegV2_Lite_256x144_infer/Portrait_PP_HumanSegV2_Lite_256x144_infer.onnx
    outputs_nodes:
    do_quantization: True
    dataset: "./Portrait_PP_HumanSegV2_Lite_256x144_infer/dataset.txt"
    output_folder: "./Portrait_PP_HumanSegV2_Lite_256x144_infer"

  • 执行转换脚本

    python tools/rknpu2/export.py
    --config_path myseg.yaml
    --target_platform rk3588

相关推荐
AlfredZhao23 天前
曾经风光无限的 Oracle DBA 已经落伍了吗?
ai·vector·embedding·onnx·hnsw·ivf
Mrs.Gril1 个月前
RKNN3588上部署 RTDETRV2
深度学习·yolo·rknn·rtdetr
IOsetting1 个月前
玩客云 OEC/OECT 笔记(2) 运行RKNN程序
rknn·rk3566·oec·oec-turbo
Milton1 个月前
玩客云 OEC/OECT 笔记(2) 运行RKNN程序
rknn·rk3566·oec·oec-turbo
雨之小2 个月前
RKNN开发环境搭建(ubuntu22.04)
rknn·rv1106
搬砖的阿wei4 个月前
ONNX:统一深度学习工作流的关键枢纽
人工智能·python·深度学习·onnx
Yongqiang Cheng5 个月前
Llama 2 Powered By ONNX
onnx·llama 2
Hi202402176 个月前
RK3588-NPU pytorch-image-models 模型编译测试
人工智能·pytorch·python·rk3588·onnx·推理
后端常规开发人员6 个月前
最好用的图文识别OCR -- PaddleOCR(2) 提高推理效率(PPOCR模型转ONNX模型进行推理)
python·ocr·onnx·paddleocr
Yao.Li6 个月前
掌握Docker命令与Dockerfile实战技巧:快速构建高效容器化应用
docker·微服务·rknn·dockerfile·docker常用命令·docker容器化