ChargingStation/chargingstation/rknn_yolo_EAI/README.md

938 B
Raw Permalink Blame History

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