AI模型转换环境搭建指南

1. 概要

模型转换环境的搭建，是将 AI 模型转换为 RKNN 格式之前必须完成的准备工作。本文档说明如何将 rknn-toolkit2 的 Docker 镜像部署到 Ubuntu 20.04 环境，并在 Docker 中启动模型转换工具运行环境。

整体流程如下。

步骤	内容
1	下载模型转换工具
2	将 Docker 镜像移动到 Ubuntu 20.04 环境
3	加载 Docker 镜像
4	进入 bash 环境，确认 RKNN 相关库是否可用

2. 下载模型转换工具

为了正常运行模型转换工具，请下载以下 Docker 镜像。

下载链接：

https://dl.dragonwake.com/download/rv1126b/AI/images/rknn-toolkit2-v2.3.2-cp38-docker.tar.gz

3. 将工具移动到 Ubuntu 20.04 环境

将下载好的 Docker 镜像移动到 Ubuntu 20.04 上的 rknn-toolkit 目录中。

参考图：将 Docker 镜像放置到 rknn-toolkit 目录中。

4. 启动模型转换工具环境

放置 Docker 镜像后，按照以下步骤启动模型转换工具环境。

4.1 打开终端

在 Docker 镜像所在目录打开终端。

4.2 加载 Docker 镜像

执行以下命令，加载模型转换工具的 Docker 镜像。

docker load --input rknn-toolkit2-v2.3.2-cp38-docker.tar.gz

4.3 进入镜像的 bash 环境

执行以下命令，进入 Docker 镜像的 bash 环境。

docker run -t -i --privileged   -v /dev/bus/usb:/dev/bus/usb   -v /home/developer/rknn-toolkit2/model_convert_test:/test   rknn-toolkit2:2.3.2-cp38 /bin/bash

执行后，可以进入容器内的 root shell。

接着输入 python，启动 Python 环境。随后确认 RKNN 相关库可以正常导入。

python
>>> import rknn

至此，模型转换工具环境搭建完成。