本仓库为SpeedDP开发平台配套的rv1126开发套件.
在linux
平台下使用convert2rknn
工具将导出的pt
模型转换为rknn
模型。
# 这里以linux子系统为例:
# 1.通过powershell打开linux子系统
wsl
# 2.在linux子系统环境运行convert2rknn
./convert2rknn.elf --env <rknn_toolkit工具包在linux子系统中的绝对路径> --model <导出模型在linux子系统中的绝对路径> --platform rv1126
注1:用户可根据需要直接使用我们打包好的rknn1.7.3.tar.gz
环境包或者参照rknn_toolkit 官方文档自行安装准备。
注2:convert2rknn
工具须搭配python3.8
进行使用,因此用户自行安装rknn_toolkit
环境时请选择python3.8
对应的安装包。
注3:convert2rknn
工具在转换RK1808/RK1806/RK3399Pro/RV1109/RV1126
等芯片平台模型时默认开启预编译选项,但高于1.7.3
版本的rknn_toolkit
只支持在线预编译,因此convert2rknn
工具不能兼容高于1.7.3
版本的rknn_toolkit
。
- 解压
gcc-arm-12.3-x86_64-arm-none-linux-gnueabihf.tar.gz
; - 进入
spdp_demo
目录并运行make.sh
编译脚本。
- 首次运行时须将
SDK
根目录lib
中的库文件拷贝至开发板上/usr/lib
目录; - 将编译生成的
spdp_demo.elf
拷贝至开发板上模型及配置文件所在目录; - 准备一张测试图片
test.jpg
,放到spdp_demo.elf
相同目录,执行spdp_demo.elf
即可得到输出结果。