RKNN AI DEMO

概述

rknn_demo 模块代码位于<SDK>/external/rknn_demo 目录下。 主要实现通过 camera 采集图像， 送到 NPU 进行处理， 并通过 minigui 显示结果。 当前 Demo 默认使用的模型为 ssd_inception_v2。 也可以直接替换为其他 SSD 的模型文件。

NPU 相关

SDK 中， 相关模型文件已经默认编译到板子中。 对应的文件宏和目录为：

#define MODEL_NAME "/usr/share/rknn_demo/ssd_inception_v2.rknn"
#define BOX_PRIORS_TXT_PATH " /usr/share/rknn_demo/box_priors.txt"
#define LABEL_NALE_TXT_PATH " /usr/share/rknn_demo/coco_labels_list.txt"

MODEL_NAME 对应模型文件， LABEL_NALE_TXT_PATH 对应结果类别列表文件， BOX_PRIORS_TXT_PATH 对应加载框权重文件。

编译

可以在 SDK 目录中， 通过命令./build.sh rknn_demo 进行模块编译， 会生成 rknn_demo 执行文件。

运行

在串口或ssh中执行如下命令可运行demo：

  1. 连接USB摄像头时

rknn_demo

  1. 连接MIPI摄像头时（购买链接)

rknn_demo --device mipi

更多

更多详情，请查阅<SDK>/docs/SoC platform related/RK1808/Rockchip RKNN_DEMO模块开发指南V0.2.pdf