CV12_ONNX转RKNN模型(谛听盒子)
创始人
2024-12-28 22:07:32
0

暂时简单整理一下:

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 RKNN  def convert_onnx_to_rknn(onnx_path,rknn_path,input_data,output_data):     #创建RKNN对象     rknn = RKNN()     rknn.config(         target_platforms='rk3588',         quantized_algorithm='mmse',         optimization_level = 2     )     #加载ONNX模型     print('loading ONNX model...')     ret = rknn.load_onnx(model=onnx_path)     if ret !=0:         print("load failed!")         return     #构建模型,进行RKNN模型转换     print('building rknn model...')     ret = rknn.build(do_quantization=False,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!')         return     print('RKNN model is successfully exported to ',rknn_path)      rknn.init_runtime()     outputs=rknn.inference(inputs=[input_data],data_format='nchw')     try:         if np.test.assert_almost_equal(outputs[0],output_data,decimal=1)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代版本不兼容rk3588),2代版本目前只有linux系统。

解决方法:搭一个虚拟环境然后用虚拟机安装rknn-toolkit2

相关内容

热门资讯

2分钟辅助挂!四川徒有辅助软件... 2分钟辅助挂!四川徒有辅助软件(辅助挂)都是是真的有挂(详细辅助解密教程);1、四川徒有辅助软件透视...
2分钟细说!wepoker辅助... 2分钟细说!wepoker辅助器免费下载(辅助挂)先前是真的有挂(详细辅助可靠技巧);建议优先通过w...
第一分钟插件!边锋辅助器小程序... 第一分钟插件!边锋辅助器小程序微信(辅助挂)扑克教程(固有存在有挂)1、很好的工具软件,可以解锁游戏...
揭秘一下"新道游ap... 揭秘一下"新道游app下载"新道游app下载(切实真的是有挂);1、操作简单,无需注册,只需要使用手...
第5分钟了解!宝宝浙江辅助脚本... 第5分钟了解!宝宝浙江辅助脚本(辅助挂)真是是有挂(详细辅助教程)关于宝宝浙江辅助脚本机制的,其中提...
第8分钟了解!欢乐达人正版脚本... 第8分钟了解!欢乐达人正版脚本(辅助挂)都是真的有挂(详细辅助总结教程);1、进入到欢乐达人正版脚本...
一分钟了解!兴动互娱辅助工具名... 一分钟了解!兴动互娱辅助工具名称(辅助挂)起初是真的有挂(详细辅助细节揭秘)是一款可以让一直输的玩家...
第八分钟有挂!新西楚辅助(辅助... 第八分钟有挂!新西楚辅助(辅助挂)力荐教程(好像有挂)1、构建自己的微扑克辅助插件;2、选择您最喜爱...
重大来袭"潮汕汇软件... 重大来袭"潮汕汇软件辅助"潮汕汇软件辅助(一直是有挂)1、超多福利:超高返利,海量正版游戏,潮汕汇软...
第一分钟了解!西元文山攻略(辅... 第一分钟了解!西元文山攻略(辅助挂)总是有挂(详细辅助攻略)准备好在西元文山攻略 ia的高塔上攀登,...