rknn_benchmark是用来测试rknn模型性能。请确保测试前将CPU/DDR/NPU频率锁定到最高频率。用法：

./rknn_benchmark xxx.rknn [input_data] [loop_count] [core_mask]

core_mask: 0: auto, 1: npu core1, 2: npu core2, 4:npu core3,

3: npu core1&2,

7: npu core1&2&3

仅RK3588支持 core mask。

例如:

./rknn_benchmark mobilenet_v1.rknn
./rknn_benchmark mobilenet_v1.rknn dog.jpg 10 3
./rknn_benchmark mobilenet_v1.rknn dog.npy 10 7
./rknn_benchmark xxx.rknn input1.npy#input2.npy

以下表示RK3566_RK3568、RK3562或RK3588。

Aarch64 Linux 示例

编译

首先导入GCC_COMPILER，例如export GCC_COMPILER=~/opt/gcc-linaro-7.5.0-2019.12-x86_64_aarch64-linux-gnu/bin/aarch64-linux-gnu ，然后执行如下命令：

./build-linux.sh -t <target> -a <arch> -b <build_type>]

# 例如: 
./build-linux.sh -t rk3588 -a aarch64 -b Release

安装

将 install/rknn_benchmark_Linux 拷贝到设备上.

如果使用Rockchip的EVB板，可以使用以下命令：

连接设备并将程序和模型传输到/userdata

adb push install/rknn_benchmark_Linux /userdata/

如果你的板子有sshd服务，可以使用scp命令或者其他方式将程序和模型传输到板子上。

运行

adb shell
cd /userdata/rknn_benchmark_Linux/

export LD_LIBRARY_PATH=./lib
./rknn_benchmark xxx.rknn

Android 示例

编译

首先导入ANDROID_NDK_PATH，例如export ANDROID_NDK_PATH=~/opts/ndk/android-ndk-r18b，然后执行如下命令：

./build-android.sh -t <target> -a <arch> [-b <build_type>]

# 例如: 
./build-android.sh -t rk3568 -a arm64-v8a -b Release

安装

connect device and push build output into /data

adb push install/rknn_benchmark_Android /data/

运行

adb shell
cd /data/rknn_benchmark_Android/

export LD_LIBRARY_PATH=./lib
./rknn_benchmark xxx.rknn