51 lines
938 B
Markdown
51 lines
938 B
Markdown
# 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
|
||
```
|
||
|
||
|