当前位置:   article > 正文

OK3588板卡上使用rknpu2加速推理yolov5和mobilenet--C语言篇(九)

rknpu2

OK3588板卡上使用rknpu2加速推理yolov5和mobilenet--C语言篇

rknpu2

RKNN SDK为带有RKNPU的芯片平台提供C语言编程接口,用于用户部署RKNN模型
把rknpu2下载到板卡,编译,然后可以运行推理程序。
这里就省掉了交叉编译的过程,都是在OK3588板卡上操作的。
我们本次的运行环境是OK3588开发板
git clone https://github.com/rockchip-linux/rknpu2

yolov5s

进入虚拟环境
conda activate ok3588
进入路径
cd rknpu2/examples/rknn_yolov5_demo/
编译程序
bash build-linux_RK3588.sh
在这里插入图片描述
运行程序:
进入编译后的install 路径
./rknn_yolov5_demo model/RK3588/yolov5s.rknn model/bus.jpg
注意yolov5s.rknn文件是在另一篇博客使用rknn-toolkit2把YOLOV5部署到OK3588上转换的rknn模型文件。
也可以使用rknn_yolov5_demo自带的rknn模型文件yolov5s-640-640.rknn
rknn模型转换是在主机Ubuntu系统上进行的,代码里面需要设置好目标平台。
在这里插入图片描述

运行结果如下图:
在这里插入图片描述
目标检测的结果有person和bus。检测框的位置和目标的置信度。

mobilenet

cd rknpu2/examples/rknn_mobilenet_demo/
编译程序
bash build-linux_RK3588.sh
在这里插入图片描述
运行程序
./rknn_mobilenet_demo model/RK3588/mobilenet_v1.rknn model/cat_224x224.jpg

运行结果:
在这里插入图片描述
TOP5类的置信度

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小小林熬夜学编程/article/detail/229572
推荐阅读
相关标签
  

闽ICP备14008679号