1. 概述
模型转换环境搭建流程如下所示: 下载模型工具
2. 下载模型转换工具
为了保证模型转换工具顺利运行，请下载网盘里“06.AI算法开发/01.rknn-toolkit2模型转换工具/rknn-toolkit2-v2.3.2/rknn-toolkit2/docker/docker_image/rknn-toolkit2-v2.3.2-cp38-docker.tar.gz”。
网盘下载链接:https://pan.baidu.com/s/1rEh1C1YDCbtsT8bH3uhirA?pwd=1234 提取码: 1234
3. 把工具移到ubuntu20.04
把下载完成的docker镜像移到我司的虚拟机ubuntu20.04的rknn-toolkit目录,如下图所示：
4. 运行模型转换工具环境
4.1 打开终端
在该目录打开终端
4.2 加载docker镜像
执行以下指令加载模型转换工具docker镜像:
docker load --input rknn-toolkit2-v2.3.2-cp38-docker.tar.gz
4.3 进入镜像bash环境
执行以下指令进入镜像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
现象如下图所示:
输入“python”加载python相关库，尝试加载rknn库，如下图环境测试成功:
至此，模型转换工具环境搭建完成。
