泰兴网站建设吧,怎么做qq靓号网站,游戏设计需要学什么专业,广州企业100强暂时简单整理一下#xff1a;
1.在边缘设备上配置相关环境。
2.配置完成后#xff0c;获取模型中间的输入输出结果#xff0c;保存为npy格式。
3.将onnx格式的模型#xff0c;以及中间输入输出文件传送到边缘设备上。
4.编写一个python文件用于转换模型格式#xff0c…暂时简单整理一下
1.在边缘设备上配置相关环境。
2.配置完成后获取模型中间的输入输出结果保存为npy格式。
3.将onnx格式的模型以及中间输入输出文件传送到边缘设备上。
4.编写一个python文件用于转换模型格式dataset作为测试可以是随便的数据。
5.将python传送到边缘设备上并运行。
注意所有文件要在同一目录下。
参考代码一
# Author:SiZhen
# Create: 2024/7/15
# Description: ONNX转RKNN
import os
import numpy as np
from rknn.api import RKNNdef convert_onnx_to_rknn(onnx_path,rknn_path,input_data,output_data):#创建RKNN对象rknn RKNN()rknn.config(target_platformsrk3588,quantized_algorithmmmse,optimization_level 2)#加载ONNX模型print(loading ONNX model...)ret rknn.load_onnx(modelonnx_path)if ret !0:print(load failed!)return#构建模型进行RKNN模型转换print(building rknn model...)ret rknn.build(do_quantizationFalse,dataset/home/DiTing/sizhen/dataset.txt)if ret !0:print(Build RKNN model failed!)#导出RKNN模型print(Exporting RKNN model...)ret rknn.export_rknn(rknn_path)if ret ! 0:print(Export RKNN failed!)returnprint(RKNN model is successfully exported to ,rknn_path)rknn.init_runtime()outputsrknn.inference(inputs[input_data],data_formatnchw)try:if np.test.assert_almost_equal(outputs[0],output_data,decimal1)is None:print(模型输出与预期一致。)except AssertionError as e :print(不一致详细信息,e)#转换模型
i 0
onnx_path r/home/DiTing/sizhen/resNet34.onnx
rknn_path onnx_path.replace(.onnx,.rknn)
input_data np.load(/home/DiTing/sizhen/input_data.npy)
output_data np.load(/home/DiTing/sizhen/output_data.npy)
convert_onnx_to_rknn(onnx_path,rknn_path,input_data,output_data)
问题及解决方法
问题一
安装失败。 解决方法这个地方谛听盒子应该是arm的版本而不是x86:Lite版本 问题二
安装成功找不到属性 解决方法用如下包替代 问题三
lite版本缺少属性 初步讨论结果lite版本应该是只包含运行环境的最好不要在盒子上进行模型转换会又慢又卡可能利特版本就是没有模型转换的功能的。
解决方法在其他环境转换。待更新可能的解决办法https://bbs.eeworld.com.cn/thread-1283146-1-1.html
问题四
在上面链接的指导进行windows环境下安装toolkit时出现如下错误 原因是没有安装Cmake。在安装Cmake后又出现如下错误 NMake Makefiles does not support platform specification, but platform x64 was specifced CMake Error: CMAKE_C_COMPILER not set, after EnableLanguage CMake Error: CMAKE_CXX_COMPILER not set, after EnableLanguage -- Configuring incomplete, errors occurred! 解决方法https://blog.csdn.net/qq_42276781/article/details/88594870
1.打开 Visual Studio Installer - 修改 - 单个组件 - 勾选 用于 CMake 的 Visual C 工具 - 修改 2.Visual Studio Installer - 修改-工作负荷 - 使用C的桌面开发 - 修改】
3.配置一下path 环境变量 计算机 - 属性 - 高级系统设置 - 环境变量 - Path - 编辑 - 加入
自己的路径\Microsoft Visual Studio\2017\Community\Common7\IDE\CommonExtensions\Microsoft\CMake\CMake\bin
后确定
重启电脑之后再次安装即可。
问题五
rknn-toolkit在windows系统没有2代版本1代版本不兼容rk35882代版本目前只有linux系统。
解决方法搭一个虚拟环境然后用虚拟机安装rknn-toolkit2
