飞凌嵌入式
直播中

筑梦者与梦同行

8年用户 179经验值
擅长:嵌入式技术
私信 关注
[活动]

【飞凌嵌入式OK3576-C开发板体验】RKNN神经网络算法开发环境搭建

一、参考文献

RKNN SDK 快速上手指南

可以通过GitHub获取到,链接:https://github.com/rockchip-linux/rknn-toolkit2

个人本帖主要描述了其实验过程,以及实验过程遇到的些许问题,与该文档有所出入。没有使用大量的篇幅重新描述实现过程,如果有同志想研究RKNN算法还是要结合RKNNSDK快速上手指南的。

二、准备开发环境

新建一个名称为 Projects 的文件夹,并将 RKNN-Toolkit2 和 RKNN Model Zoo 仓库存放至该目录下
图片31.png图片32.png

三、安装RKNN-Toolkit2环境

3.1、安装Conda

Conda是一个开源的软件包管理系统和环境管理系统,主要用于安装、更新、删除软件包及其依赖关系,并允许用户在不同环境之间轻松切换。用于后续安装Python等插件。在深度神经算法中使用的频率非常高的一款软件。
图片33.png图片34.png
图片35.png

图片36.png

图片37.png

图片38.png

图片39.png

3.2、使用PIP命令安装依赖库和Toolkit2 工具
图片40.png图片41.png
图片42.png

图片43.png

图片44.png

四、RKNN 程序转换

进入 rknn_model_zoo/examples/yolov5/model 目录,运行 download_model.sh 脚本,该脚本

将下载一个可用的 YOLOv5 ONNX 模型,并存放在当前 model 目录下,参考命令如下:
图片45.png图片46.png
图片47.png

图片48.png

安装COCO数据集,在深度神经网络算法中,模型的训练离不开大量的数据集,数据集用于神经网络模型的训练。

图片49.png

更多回帖

发帖
×
20
完善资料,
赚取积分