认知智能
ced33f5abc
|
||
|---|---|---|
| .. | ||
| include | ||
| install | ||
| lib | ||
| libs | ||
| model | ||
| src | ||
| CMakeLists.txt | ||
| README.md | ||
| build.sh | ||
README.md
rknn_yolo_EAI_pic
简要说明
用于跑EAI提供仓库训练出来的yolov5模型的图片检测例程。 用法。/rknn_yolo_EAI_pic model picture
需注意运行该例程需要将npu驱动更新为1.7.3版本
准备工作
根据EAI给出的教程训练、转化得到rknn模型
编译(build)
根据设备，调整 RK1808_TOOL_CHAIN/RV1109_TOOL_CHAIN 参数路径及 GCC_COMPILER 参数，执行下面命令进行编译
./build.sh
- rv1126/rv1109使用同一套编译工具
安装(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_pic yolov5 ./yolov5.rknn ./model/dog.jpg