#AI挑战营第二站#从零开始教你使用RKNN-Toolkit进行模型转换

Alex_Jun

#AI挑战营第二站#从零开始教你使用RKNN-Toolkit进行模型转换 [复制链接]

<i class="pstatus"> 本帖最后由 Alex_Jun 于 2024-5-9 20:54 编辑 </i>

# RKNN-Toolkit环境部署与使用
RKNN-Toolkit是一款软件开发套件，可在PC和瑞芯微NPU平台（RK1808/RK1806/RK3399Pro/RV1109/RV1126）上为用户提供模型转换、推理和性能评估。
想要做好一件事，我们首先得要了解我们所需要的东西，正所谓工欲善其事必先利其器。
## 一、 下载RKNN-Toolkit

```
RKNN-Toolkit的github地址为：
https://github.com/rockchip-linux/rknn-toolkit
```
我们所要部署的开发板为幸狐RV1106开发板，
因此我们应该参考

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807204_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image.png</strong> <em class="xg1">(219.33 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807204" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807204&handlekey=savephoto_807204">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

因此我们点击链接跳转到rknn-toolkit2的链接去
```
https://github.com/rockchip-linux/rknn-toolkit2
```

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807205_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-2.png</strong> <em class="xg1">(12.37 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807205" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807205&handlekey=savephoto_807205">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

但是我们发现它以及不再维护和更新了，不过给了我们新的地址，我们点击跳转到新的地址去：
```
https://github.com/airockchip/rknn-toolkit2/
```
我用网页给大家翻译了一下，方便我们查看：

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807206_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-3.png</strong> <em class="xg1">(71.24 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807206" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807206&handlekey=savephoto_807206">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

具体的东西我们可以不用很了解，但是我们可以看到这句话：

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807207_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-5.png</strong> <em class="xg1">(6.97 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807207" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807207&handlekey=savephoto_807207">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

这不就是我们需要的功能吗，所以我们需要下载这个仓库。
但是要注意，如果你下载了RKNN-Toolkit可能会出问题，因为它说两个不相容。

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807208_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-6.png</strong> <em class="xg1">(13.14 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807208" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807208&handlekey=savephoto_807208">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

## 二、寻找教程
如果你不想寻找教程可以直接跳过这一步，我会将教程里面用到的东西贴出来，可以直接跳转**3.使用教程**，但是如果你想要了解这个工具，那么你一定要找到教程。
那么，在这个页面翻一翻，瞧我们发现了什么，更多的例程，有了例程不就可以来帮助我们理解和使用工具了吗?

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807209_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-27.png</strong> <em class="xg1">(13.23 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807209" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807209&handlekey=savephoto_807209">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

我们点击链接进去：
```
https://github.com/airockchip/rknn_model_zoo
```
在这个页面也翻一翻，找到了什么，**Quick Start**

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807210_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-28.png</strong> <em class="xg1">(164.6 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807210" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807210&handlekey=savephoto_807210">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

我们点进这个链接看一看：
```
https://github.com/airockchip/rknn-toolkit2/tree/master/doc
```
我相信你一定看到了很多版本的教程文件，我们应该选择RV1106对应的且后缀为CN的.pdf教程，如果你想学英语也可以选择EN结尾的。

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807211_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-29.png</strong> <em class="xg1">(235.18 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807211" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807211&handlekey=savephoto_807211">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

下载文件打开，先看封面再看目录，好的确实是我们要找的教程了，没错了：

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807212_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-32.png</strong> <em class="xg1">(12.95 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807212" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807212&handlekey=savephoto_807212">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807213_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-31.png</strong> <em class="xg1">(83.33 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807213" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807213&handlekey=savephoto_807213">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

## 三、使用教程
教程找到了，我们又没有开发板，所以我们直接从**3 准备开发环境**开始寻找有用的东西。

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807214_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-34.png</strong> <em class="xg1">(71.42 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807214" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807214&handlekey=savephoto_807214">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

好的，看完了，已经学会了，我们可以开始了：
### 1.下载仓库
这一步直接照着他的做可以，文件夹的名字可以修改，我这里使用的文件名是**code**，这个不影响后续操作。

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807215_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-35.png</strong> <em class="xg1">(118.29 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807215" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807215&handlekey=savephoto_807215">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807216_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-36.png</strong> <em class="xg1">(65.83 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807216" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807216&handlekey=savephoto_807216">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

### 2.安装 RKNN-Toolkit2 环境

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807217_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-37.png</strong> <em class="xg1">(156.5 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807217" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807217&handlekey=savephoto_807217">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

```
conda -V
```
查看是否安装，虽然这一步我安装了也是这个结果，但是看一看，一般的新的虚拟机都没有安装，我们直接下一步就行。

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807218_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-38.png</strong> <em class="xg1">(35.31 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807218" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807218&handlekey=savephoto_807218">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

下载安装包链接，直接复制到命令行运行就行，这个会自己下载的：
```
wget -c https://mirrors.bfsu.edu.cn/anaconda/miniconda/Miniconda3-latest-Linux-x86_64.sh
```
安装conda，也是照做就行，反正都是新手教程。
```
chmod 777 Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh
```
**使用 Conda 创建 Python 环境**

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807219_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-39.png</strong> <em class="xg1">(77.34 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807219" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807219&handlekey=savephoto_807219">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

在命令行输入：
```
source ~/miniconda3/bin/activate
```

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807220_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-40.png</strong> <em class="xg1">(32.13 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807220" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807220&handlekey=savephoto_807220">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

通过以下命令创建名称为 toolkit2 的 Python 3.8 环境：
```
conda create -n toolkit2 python=3.8
```
我已经创建过了就不再创建了，接下来我们激活 toolkit2 环境， 后续将在此环境中安装 RKNN-Toolkit2：
```
conda activate toolkit2
```

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807221_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-43.png</strong> <em class="xg1">(42.45 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807221" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807221&handlekey=savephoto_807221">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

**安装依赖库和 RKNN-Toolkit2**

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807222_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-44.png</strong> <em class="xg1">(121.2 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807222" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807222&handlekey=savephoto_807222">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

我们根据自己的路径来，只要到packages的上一级目录就行：

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807223_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-45.png</strong> <em class="xg1">(95.88 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807223" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807223&handlekey=savephoto_807223">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

接下来就是安装依赖了，我们上面创建的python环境是3.8的，教程上说根据python版本选择依赖文件，因此我们选择**requirements_cp38-2.0.0b0.txt**文件。
*为什么我这里只有38和39？因为不想占内存所有，我把其他的都删了，剩个39懒得删了*。

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807224_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-46.png</strong> <em class="xg1">(58.81 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807224" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807224&handlekey=savephoto_807224">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

所以我们输入的命令为：
```
pip install -r packages/requirements_cp38-2.0.0b0.txt
```
接下来就是等它下载...
如果出现**Read Time OUT**错误怎么办？
再执行一遍就行，我第一遍也超时了，多试几次就行，或者可以上网百度conda pip 换源，但是这里还挺快，所以我没换了。

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807225_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-8.png</strong> <em class="xg1">(125.11 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807225" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807225&handlekey=savephoto_807225">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

然后来安装工具了，我们还是根据python版本选择38的那个：
```
pip install packages/rknn_toolkit2-2.0.0b09bab5682-cp38-cp38-linux_x86_64.whl
```

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807226_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-9.png</strong> <em class="xg1">(147.72 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807226" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807226&handlekey=savephoto_807226">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

查看python是否安装成功

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807227_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-47.png</strong> <em class="xg1">(72.29 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807227" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807227&handlekey=savephoto_807227">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

输入以下命令，没报错就是对的：
```
# 进入 Python 交互模式
python
# 导入 RKNN 类
from rknn.api import RKNN
```

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807228_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-10.png</strong> <em class="xg1">(142.57 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807228" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807228&handlekey=savephoto_807228">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

那么，到这里我们的开发环境就搭建的差不多了，至少转换模型所需要的环境已经够了，后面的等板子到了再说吧。
### 3.准备转换

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807229_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-48.png</strong> <em class="xg1">(35.86 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807229" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807229&handlekey=savephoto_807229">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807230_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-15.png</strong> <em class="xg1">(58.67 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807230" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807230&handlekey=savephoto_807230">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807231_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-16.png</strong> <em class="xg1">(83.79 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807231" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807231&handlekey=savephoto_807231">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

在运行之前我们需要进行一点改变，因为我们使用的不是yolo模型，而是MNIST训练的Pytorch模型，与网上的yolov5模型不同，这里我给一个网站：
```
https://netron.app/
```

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807232_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-23.png</strong> <em class="xg1">(4.94 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807232" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807232&handlekey=savephoto_807232">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

这是别人yolov5模型的输入：

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807233_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-22.png</strong> <em class="xg1">(27.61 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807233" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807233&handlekey=savephoto_807233">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

我相信你肯定发现不同了，我们这个输入是**input.1**我现在也不知都这个是为什么，但是跟yolo肯定是有点区别的。也会对后续我们的操作造成一点不同。
我们需要修改
```
rknn_model_zoo-2.0.0/examples/yolov5/python/convert.py
```
中的四十四行：
```python
  # rknn.config(mean_values=[[0, 0, 0]], std_values=[
  #     [255, 255, 255]], target_platform=platform)
  #修改为
  rknn.config(target_platform=platform)
```

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807234_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-19.png</strong> <em class="xg1">(264.3 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807234" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807234&handlekey=savephoto_807234">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

接下来就是模型转换了，我们输入这里的命令应该是
```
#注意my_model为我们模型的名字，将其放到model路径下
python convert.py ../model/my_model.onnx rv1106 i8 ../model/my_model.rknn
```

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807235_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-25.png</strong> <em class="xg1">(74.73 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807235" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807235&handlekey=savephoto_807235">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

<ignore_js_op>

<div class="tip tip_4 aimg_tip" id="aimg_807236_menu" style="position: absolute; display: none" disautofocus="true">
<div class="xs0">
<p><strong>image-24.png</strong> <em class="xg1">(106.65 KB, 下载次数: 0)</em></p>
<p>
<a href="javascript:;" class="loginf" >下载附件</a>

<a href="javascript:;" onclick="showWindow(this.id, this.getAttribute('url'), 'get', 0);" id="savephoto_807236" url="home.php?mod=spacecp&ac=album&op=saveforumphoto&aid=807236&handlekey=savephoto_807236">保存到相册</a>

</p>

</div>
<div class="tip_horn"></div>
</div>

</ignore_js_op>

执行完命令就会生成.rknn模型了。

---

### 4.代码解读
这是我们使用到的官方的代码，我来浅浅的解读一下。
```python
import sys
from rknn.api import RKNN

DATASET_PATH = '../../../datasets/MNIST/mnist_subset_20.txt'# 这个就是我们测试数据的路径，用来量化模型
DEFAULT_RKNN_PATH = '../model/yolov5.rknn'    # 这是默认的模型输出路径和名字，我们命令行里面输入流参数就不会使用这个了
DEFAULT_QUANT = True  # 这个就是量化，RKNN库会根据数据集文件（就是我们上面那个路径）中的数据，自动调整量化结果以获得最佳性能。
# 这就是对参数进行一个处理，咱不管他
def parse_arg():
  if len(sys.argv) < 3:
  print("Usage: python3 {} onnx_model_path [platform] [dtype(optional)] [output_rknn_path(optional)]".format(sys.argv[0]))
  print("   platform choose from [rk3562,rk3566,rk3568,rk3588,rk1808,rv1109,rv1126]")
  print("   dtype choose from [i8, fp] for [rk3562,rk3566,rk3568,rk3588]")
  print("   dtype choose from [u8, fp] for [rk1808,rv1109,rv1126]")
  exit(1)

model_path = sys.argv[1]
  platform = sys.argv[2]

do_quant = DEFAULT_QUANT
  if len(sys.argv) > 3:
  model_type = sys.argv[3]
  if model_type not in ['i8', 'u8', 'fp']:
    print("ERROR: Invalid model type: {}".format(model_type))
    exit(1)
  elif model_type in ['i8', 'u8']:
    do_quant = True
  else:
    do_quant = False

if len(sys.argv) > 4:
  output_path = sys.argv[4]
  else:
  output_path = DEFAULT_RKNN_PATH

return model_path, platform, do_quant, output_path

if __name__ == '__main__':
  model_path, platform, do_quant, output_path = parse_arg()

# 创建一个rknn模型，verbose=False为不打印输出信息
  rknn = RKNN(verbose=False)

# 这就是我们的预处理配置，设置目标平台，参数从命令行得到
  print('--> Config model')  
  rknn.config(target_platform=platform)
  print('done')

# 加载我们的onnx模型
  print('--> Loading model')
  ret = rknn.load_onnx(model=model_path)
  if ret != 0:
  print('Load model failed!')
  exit(ret)
  print('done')

# 根据onnx模型转化为rknn模型
  print('--> Building model')
  ret = rknn.build(do_quantization=do_quant, dataset=DATASET_PATH)
  if ret != 0:
  print('Build model failed!')
  exit(ret)
  print('done')

# 将训练好的模型导出到文件中
  print('--> Export rknn model')
  ret = rknn.export_rknn(output_path)
  if ret != 0:
  print('Export rknn model failed!')
  exit(ret)
  print('done')

# 释放资源
  rknn.release()
```
看完代码我们发现有用的就那么几行，所以我们这一站也算是比较简单，但是前面环境配置比较费时间，网上的教程我找了一些后来还是打算自己弄，弄完发现也不是很难对吧。

## 总结
那么我们这一站的任务就完成了，但是目前板子还没到无法验证我们转化后的模型性能，所以后续打算将模型部署到 RV1106 Linux 开发板上再来决定是否需要更换模型。
让我们一起期待板子的到来，来进行后续的测试吧！
最后贴上转化前后的模型吧：

freebsder

很详细，谢谢分享！收藏待查

#AI挑战营第二站#从零开始教你使用RKNN-Toolkit进行模型转换 [复制链接]

最新回复