tickley / rknn_yolov5_Multithreading

3 stars 0 forks source link

rknn_yolov5_Multithreading 在RK官方的Yolo-v5 dome修改,加入多线程

导出rknn模型步骤

请参考 https://github.com/airockchip/rknn_model_zoo/tree/main/models/CV/object_detection/yolo

注意事项

  1. 使用rknn-toolkit2版本大于等于1.4.0。
  2. 切换成自己训练的模型时,请注意对齐anchor等后处理参数,否则会导致后处理解析出错。
  3. 官网和rk预训练模型都是检测80类的目标,如果自己训练的模型,需要更改include/postprocess.h中的OBJ_CLASS_NUM以及NMS_THRESH,BOX_THRESH后处理参数。
  4. demo需要librga.so的支持,编译使用请参考 https://github.com/airockchip/librga
  5. 由于硬件限制,该demo的模型默认把 yolov5 模型的后处理部分,移至cpu实现。本demo附带的模型均使用relu为激活函数,相比silu激活函数精度略微下降,性能大幅上升。

Aarch64 Linux Demo

编译

支持rk3566 rk3568 rk3588根据指定平台修改 build-linux_<TARGET_PLATFORM>.sh中的交叉编译器所在目录的路径 TOOL_CHAIN,例如修改成

export TOOL_CHAIN=~/opt/tool_chain/gcc-9.3.0-x86_64_aarch64-linux-gnu/host

然后执行:

./build-linux_<TARGET_PLATFORM>.sh

推送执行文件到板子

将 install/rknn_yolov5_demo_Linux 拷贝到板子的/userdata/目录.

adb push install/rknn_yolov5_demo_Linux /userdata/

运行

adb shell
cd /userdata/rknn_yolov5_demo_Linux/

export LD_LIBRARY_PATH=./lib
//打开/dev/video11 mipi摄像头
./rknn_yolov5_demo model/<TARGET_PLATFORM>/yolov5s-640-640.rknn 11

//打开mp4文件
./rknn_yolov5_demo model/<TARGET_PLATFORM>/yolov5s-640-640.rknn ./720p60hz.mp4

Note: Try searching the location of librga.so and add it to LD_LIBRARY_PATH if the librga.so is not found on the lib folder. Using the following commands to add to LD_LIBRARY_PATH.

export LD_LIBRARY_PATH=./lib:<LOCATION_LIBRGA.SO>

RTSP推送:

//camera -> yolov5 -> h264 -> rtsp 仅支持打开YUV数据的camera ./rknn_H246_rtsp_demo model//yolov5s-640-640.rknn 11


- 接收

在vlc中打开 rtsp://192.168.2.97/live/main_stream 串流 ip修改为本机


## 视频流Demo运行命令参考如下:
- h264视频

./rknn_yolov5_video_demo model//yolov5s-640-640.rknn xxx.h264 264

注意需要使用h264码流视频,可以使用如下命令转换得到:

ffmpeg -i xxx.mp4 -vcodec h264 xxx.h264


- h265视频

./rknn_yolov5_video_demo model//yolov5s-640-640.rknn xxx.hevc 265

注意需要使用h265码流视频,可以使用如下命令转换得到:

ffmpeg -i xxx.mp4 -vcodec hevc xxx.hevc

- rtsp视频流

./rknn_yolov5_video_demo model//yolov5s-640-640.rknn 265



### 注意

- 需要根据系统的rga驱动选择正确的librga库,具体依赖请参考: https://github.com/airockchip/librga
- **rtsp 视频流Demo仅在Linux系统上支持,Android上目前还不支持**
- 视频流输入的h264名称不能为"out.h264",会被覆盖