|
||
---|---|---|
.. | ||
include | ||
install | ||
lib | ||
libs | ||
model | ||
src | ||
CMakeLists.txt | ||
README.md | ||
build.sh |
README.md
rknn_yolo_EAI_pic
简要说明
用于跑EAI提供仓库训练出来的yolov5模型的视频检测例程。 集成了从串口读取红外温度数据,到达阈值后报警并运行rknn模型
需注意运行该例程需要将npu驱动更新为1.7.3版本
准备工作
根据EAI给出的教程训练、转化得到rknn模型 有两类检测模型:火焰检测和烟雾检测
编译(build)
根据设备,调整 GCC_COMPILER 参数,执行下面命令进行编译
./build.sh
安装(install)
通过adb连接设备并将编译结果推送至板端,执行以下命令推送,这里板端默认使用/userdata路径
adb push install/rknn_yolo_demo /userdata/
将前面准备好的RKNN模型推送至板端,这里假设模型名字为 yolov5s_u8.rknn
adb push ./yolov5s_u8.rknn /userdata/rknn_yolo_demo/model/yolov5s_u8.rknn
单图测试执行
./rknn_yolo_EAI