onnx模型转换到rknn脚本

from rknn.api import RKNN

ONNX_MODEL = './onnx_models/yolov5s_rm_transpose.onnx'

platform="rk1808"

platform = "rv1109"

RKNN_MODEL = 'yolov5s_relu_{}_out_opt.rknn'.format(platform)

if name == 'main':

add_perm = False # 如果设置成True,则将模型输入layout修改成NHWC

Create RKNN object

rknn = RKNN(verbose=True)

pre-process config

print('--> config model')

rknn.config(batch_size=1, mean_values=[[0, 0, 0]], std_values=[[255, 255, 255]], reorder_channel='0 1 2', target_platform=[platform],

force_builtin_perm=add_perm, output_optimize=1)

print('done')

Load tensorflow model

print('--> Loading model')

ret = rknn.load_onnx(model=ONNX_MODEL)

if ret != 0:

print('Load resnet50v2 failed!')

exit(ret)

print('done')

Build model

print('--> Building model')

ret = rknn.build(do_quantization=True, dataset='./dataset.txt')

if ret != 0:

print('Build resnet50 failed!')

exit(ret)

print('done')

rknn.export_rknn_precompile_model(RKNN_MODEL)

rknn.export_rknn(RKNN_MODEL)

rknn.release()

相关推荐
泰迪智能科技011 小时前
高校深度学习视觉应用平台产品介绍
人工智能·深度学习
盛派网络小助手2 小时前
微信 SDK 更新 Sample,NCF 文档和模板更新,更多更新日志,欢迎解锁
开发语言·人工智能·后端·架构·c#
算法小白(真小白)2 小时前
低代码软件搭建自学第二天——构建拖拽功能
python·低代码·pyqt
唐小旭2 小时前
服务器建立-错误:pyenv环境建立后python版本不对
运维·服务器·python
007php0072 小时前
Go语言zero项目部署后启动失败问题分析与解决
java·服务器·网络·python·golang·php·ai编程
Eric.Lee20212 小时前
Paddle OCR 中英文检测识别 - python 实现
人工智能·opencv·计算机视觉·ocr检测
cd_farsight2 小时前
nlp初学者怎么入门?需要学习哪些?
人工智能·自然语言处理
AI明说2 小时前
评估大语言模型在药物基因组学问答任务中的表现:PGxQA
人工智能·语言模型·自然语言处理·数智药师·数智药学
Chinese Red Guest2 小时前
python
开发语言·python·pygame
Focus_Liu2 小时前
NLP-UIE(Universal Information Extraction)
人工智能·自然语言处理