模型转换环境搭建

1. 概述

模型转换环境搭建流程如下所示:

../_images/rknn-toolkit01.png

2. 下载模型转换工具

3. 把工具移到ubuntu18.04

把下载完成的docker镜像移到我司的虚拟机ubuntu18.04的rknn-toolkit目录,如下图所示:

../_images/rknn-toolkit02.png

4. 运行模型转换工具环境

4.1 打开终端

在该目录打开终端:

../_images/rknn-toolkit03.png

4.2 加载docker镜像

执行以下指令加载模型转换工具docker镜像:

docker load --input /home/Water-drop/rknn-toolkit/rknn-toolkit-1.7.1-docker.tar.gz

4.3 进入镜像bash环境

执行以下指令进入镜像bash环境:

docker run -t -i --privileged -v /dev/bus/usb:/dev/bus/usb rknn-toolkit:1.7.1 /bin/bash

现象如下图所示:

../_images/rknn-toolkit04.png

4.4 测试环境

输入“python”加载python相关库,尝试加载rknn库,环境测试成功(如下图),至此,模型转换工具环境搭建完成。

../_images/rknn-toolkit05.png