1. 概述

模型轉(zhuǎn)換環(huán)境搭建流程如下所示: 下載模型工具

2. 下載模型轉(zhuǎn)換工具
為了保證模型轉(zhuǎn)換工具順利運(yùn)行，請下載網(wǎng)盤里“06.AI算法開發(fā)/01.rknn-toolkit2模型轉(zhuǎn)換工具/rknn-toolkit2-v2.3.2/rknn-toolkit2/docker/docker_image/rknn-toolkit2-v2.3.2-cp38-docker.tar.gz”。
網(wǎng)盤下載鏈接:https://pan.baidu.com/s/1rEh1C1YDCbtsT8bH3uhirA?pwd=1234 提取碼: 1234

3. 把工具移到ubuntu20.04

把下載完成的docker鏡像移到我司的虛擬機(jī)ubuntu20.04的rknn-toolkit目錄,如下圖所示：

4. 運(yùn)行模型轉(zhuǎn)換工具環(huán)境

4.1 打開終端

在該目錄打開終端

4.2 加載docker鏡像

執(zhí)行以下指令加載模型轉(zhuǎn)換工具docker鏡像:

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

4.3 進(jìn)入鏡像bash環(huán)境

執(zhí)行以下指令進(jìn)入鏡像bash環(huán)境:

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

現(xiàn)象如下圖所示:

輸入“python”加載python相關(guān)庫，嘗試加載rknn庫，如下圖環(huán)境測試成功:

至此，模型轉(zhuǎn)換工具環(huán)境搭建完成。