Skip to main content

RKNN RockX 模型测试

本文档将指导您在 RV1126/RV1109 AI 摄像头上进行模型测试。测试涉及两个主要应用：

  • rkmedia_vi_rknn_venc_rtsp_test：基于 RKNN 模型的 RTSP 流媒体测试
  • rkmedia_vi_rockx_venc_rtsp_test：基于 RockX 模型的 RTSP 流媒体测试

准备工作​

在进行测试之前，您需要准备以下文件：

  • 模型文件：rknn_model.zip

下载地址

内容如下:

10_file.png

上传模型文件到设备​

将下载的模型文件上传到设备的 /oem/usr/share/ 目录下：

adb push rtsp-nn.cfg rknn_model /oem/usr/share/

10_push.png

设置环境变量​

在运行测试之前，您需要设置环境变量 以及 停止 IPC 网络摄像头 以确保程序能够正确加载所需的库：

/etc/init.d/S98_lunch_init stop
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/oem/usr/lib/:/vendor/lib/

10_export.png

运行测试​

1. 基于 RKNN 模型的测试​

运行以下命令以启动基于 RKNN 模型的 RTSP 流媒体测试：

/oem/usr/bin/rkmedia_vi_rknn_venc_rtsp_test \
    -a /etc/iqfiles/ \
    -c /oem/usr/share/rtsp-nn.cfg \
    -b /oem/usr/share/rknn_model/box_priors.txt \
    -l /oem/usr/share/rknn_model/coco_labels_list.txt \
    -p /oem/usr/share/rknn_model/ssd_mobilenet_v1_rv1109_rv1126.rknn

10_rknn.png

10_rknn2.png

2. 基于 RockX 模型的测试​

运行以下命令以启动基于 RockX 模型的 RTSP 流媒体测试：

/oem/usr/bin/rkmedia_vi_rockx_venc_rtsp_test \
  -a /etc/iqfiles/ \       # ISP调优文件路径
  -c /oem/usr/share/rtsp-nn.cfg \  # 配置文件（必须）
  -l /oem/usr/share/rknn_model/coco_labels_list.txt \  # 标签文件
  -p /oem/usr/share/rknn_model/ssd_inception_v2_rv1109_rv1126.rknn \  # 模型文件
  -t 0.5  # 调整置信度阈值

10_rockx.png

10_rockx2.png

查看流媒体效果​

测试运行后，您可以通过以下 RTSP 流地址查看测试效果：

  • rtsp://[IP]/live/sub_stream
  • rtsp://[IP]:1935/live/substream

通过以上步骤，您可以顺利在 RV1126/RV1109 AI 摄像头上完成模型测试，并查看实时流媒体效果。

常用命令​

启动推流
. /oem/usr/bin/rkmedia_vi_venc_rtsp_test -a /etc/iqfiles/ -w 1920 -h 1080 -d rkispp_scale0 &

取帧率
rkmedia_vi_get_frame_test  -a /etc/iqfiles/ -w 1920 -h 1080 -d rkispp_scale0 -o /tmp/1080p.nv12 -c 10

录视频
rkmedia_vi_venc_test  -a /etc/iqfiles/ -w 1920 -h 1080 -d rkispp_scale0 -o /tmp/1080p.h264 -c 100