本次实验说明 将TB-RK1808S0算力棒插入到Win10操作系统的电脑。通过ssh连接到1808S0的linux系统。在平台上部署基于RKNN-Toolkit导出的模型。具体步骤如下: 一、配置宿主机的IP地址 ping 192.168.180.8 成功ping 通则说明没有问题。rk1808s0的IP地址默认是192.168.180.8。 二、配置NAT网络,让rk1808s0可以上网 网络属性-》更改适配器设置
WLAN属性-》共享-》以太网2 重新设置以太网2的IP地址为192.168.180.1 ssh 登录192.168.180.8 然后sudo ping 如果能够ping 通则代表网络配置成功。 三、登录并配置主从模式
配置完成之后,就可以通过ssh来连接到rk1808S0四、Ubuntu18.04安装RKNN-Toolkit 安装docker容器 通过docker方式安装,下载docker 镜像 导入rknn镜像 sudo docker load --input rknn-toolkit-1.3.2-docker.tar.gz 启动镜像 sudo docker run -it --privileged -v /home/nemo/engine/rknn/:/opt/rknn rknn-toolkit:1.3.2 /bin/bash 五、利用RKNN-Toolkit转换模型 rknn = RKNN() #初始化rknn对象 # pre-process config print('--> config model') rknn.config(channel_mean_value='103.94 116.78 123.68 58.82', reorder_channel='0 1 2') #设置参数mean scale 通道顺序 print('done') # Load tensorflow model print('--> Loading model') ret = rknn.load_tflite(model='./mobilenet_v1.tflite') #载入tflite 模型 if ret != 0: print('Load mobilenet_v1 failed!') exit(ret) print('done') # Build model print('--> Building model') ret = rknn.build(do_quantization=True, dataset='./dataset.txt')#模型量化可以做也可以不做 if ret != 0: print('Build mobilenet_v1 failed!') exit(ret) print('done') # Export rknn model print('--> Export RKNN model') ret = rknn.export_rknn('./mobilenet_v1.rknn')#导出模型 if ret != 0: print('Export mobilenet_v1.rknn failed!') exit(ret) print('done') 六、ssh连接登录到rk1808S0,基于SDK开发模型推理 sudo dnf update -y 在线升级固件到最新版本
原作者:RunningJiang
|