jnulzl / YOLOV5_RK1126

yolov5 rk1126 cpp code

ONNX模型RK1126平台部署(以yolov5为例)

环境配置

• Ubuntu18.04/Ubuntu20.04

• cmake, git, make等等,根据实际情况缺什么安装什么

• 交叉编译工具gcc-arm-9.2-2019.12-x86_64-arm-none-linux-gnueabihf

模型量化

见models/detect_yolov5_onnx/README.md

构建

git clone https://github.com/jnulzl/YOLOV5_RK1126
cd YOLOV5_RK1126
mkdir build && cd build
export CROSS_COMPILER_ROOT = 'YOUR CROSS COMPILER ROOT'
cmake ../ -DCMAKE_BUILD_TYPE=Release -DCMAKE_CXX_COMPILER=$CROSS_COMPILER_ROOT/bin/arm-none-linux-gnueabihf-g++ -DCMAKE_C_COMPILER=$CROSS_COMPILER_ROOT/bin/arm-none-linux-gnueabihf-gcc

输出产物在YOLOV5_RK1126/bin/Linux下面，拷贝到板子跑即可。

將模型转化为buffer直接编译到库或者可执行文件中

见examples/README.md