RKNN Toolkit2 工具的使用

RKNN Toolkit2 是由瑞芯微电子 (Rockchip) 开发的一套用于深度学习模型优化和推理的工具。它主要面向在瑞芯微SoC上进行AI应用开发,但也可以用于PC平台进行模型的转换、量化、推理等操作。它支持将多种深度学习框架的模型(如Caffe, TensorFlow, PyTorch等)转换为RKNN格式。

创建虚拟环境

使用miniconda或者Anaconda创建一个虚拟环境,其中miniconda 是 Anaconda 的轻量版。为了减少内存,我们采用miniconda进行开发。

下载链接:

复制代码
​wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh​

安装miniconda,执行以下命令

复制代码
./Miniconda3-latest-Linux-x86_64.sh​

然后回车,一直往下滑动看完 license,最后输入 yes 后,继续按下回车, 然后进入安装,安装完成后输入 yes 初始化 anaconda3。

创建虚拟环境,执行以下命令

复制代码
conda create -n Toolkit2 python=3.8.12    //创建一个名为Toolkit2 的虚拟环境,Python的版本为3.8.12
conda activate Toolkit2        //进入虚拟环境
复制代码
conda env list   //查看虚拟环境列表
conda deactivate   //退出虚拟环境

Toolkit2的安装

安装依赖

复制代码
sudo apt update

sudo apt-get install python3-dev python3-pip python3.8-venv gcc

sudo apt-get install libxslt1-dev zlib1g-dev libglib2.0 libsm6 \libgl1-mesa-glx libprotobuf-dev gcc

安装 RKNN-Toolkit2

复制代码
mkdir project-Toolkit2 && cd project-Toolkit2

拉取RKNN-Toolkit2

复制代码
git clone https://gitee.com/LubanCat/lubancat_ai_manual_code.git
cd lubancat_ai_manual_code/dev_env/rknn_toolkit2

配置pip源

复制代码
pip3 config set global.index-url https://mirror.baidu.com/pypi/simple

进入到 lubancat_ai_manual_code/dev_env/rknn_toolkit2/doc目录,安装依赖库

复制代码
pip3 install numpy
pip3 install -r requirements_cp38-1.5.0.txt

建议在安装需求之前,sudo vim requirements_cp38-1.5.0.txt,numpy==1.19.5注释掉,否则可能导致依赖安装shibai

安装 rknn_toolkit2

复制代码
pip3 install rknn_toolkit2-1.5.0+1fa95b5c-cp38-cp38-linux_x86_64.whl

检查是否安装成功

输入 quit() 或者使用快捷键 Ctrl+D 退出

相关推荐
爱打球的白师傅17 分钟前
python机器学习工程化demo(包含训练模型,预测数据,模型列表,模型详情,删除模型)支持线性回归、逻辑回归、决策树、SVC、随机森林等模型
人工智能·python·深度学习·机器学习·flask·逻辑回归·线性回归
MediaTea1 小时前
Python 第三方库:TensorFlow(深度学习框架)
开发语言·人工智能·python·深度学习·tensorflow
Joker-Tong1 小时前
大模型数据洞察能力方法调研
人工智能·python·agent
B站计算机毕业设计之家1 小时前
基于Python+Django+双协同过滤豆瓣电影推荐系统 协同过滤推荐算法 爬虫 大数据毕业设计(源码+文档)✅
大数据·爬虫·python·机器学习·数据分析·django·推荐算法
逻极1 小时前
Webhook 全解析:事件驱动时代的实时集成核心技术
python·web
程序员三藏1 小时前
一文了解UI自动化测试
自动化测试·软件测试·python·selenium·测试工具·职场和发展·测试用例
极客代码1 小时前
第七篇:深度学习SLAM——端到端的革命--从深度特征到神经辐射场的建图新范式
人工智能·python·深度学习·计算机视觉·slam·回环检测·地图构建
有Li1 小时前
面向超声半监督分割的类别特异性无标记数据风险最小化|文献速递-文献分享
人工智能·深度学习·计算机视觉
larance2 小时前
python中的鸭子类型
开发语言·python
陈辛chenxin2 小时前
【大数据技术04】数据可视化
大数据·python·信息可视化