1.简介
基于RKNPU2 SDK 1.6.0版的安卓YOLOv5演示应用程序,选择图片进行对象检测并显示识别结果。
GitHub源码地址:https://github.com/shiyinghan/rknn-android-yolov5
2.实现过程
参考RKNN官方库RKNN Model Zoo提供的YOLOv5对象检测demo,该demo是二进制的可执行程序,没有整合成app,但是代码结构比较清晰,整合成安卓应用,可以熟悉一下流程。
App相关的功能,例如加载asset下面的rknn模型文件等,参考了ncnn-android-yolov5软件库。
调用模型推理和后处理逻辑参考rknn_model_zoo/examples/yolov5/cpp下面的demo源码。
3.运行效果
先选择图片,然后点击识别,效果如下:
4.注意事项
运行该demo之前需要保证开发板端已经配置好了RKNN环境,/vendor/bin/rknn_server和/vendor/lib64/librknnrt.so的版本都是1.6.0。
RKNN官方已经有使用摄像头的实时数据进行yolov5对