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()

相关推荐
黎燃6 小时前
短视频平台内容推荐算法优化:从协同过滤到多模态深度学习
人工智能
TF男孩7 小时前
ARQ:一款低成本的消息队列,实现每秒万级吞吐
后端·python·消息队列
飞哥数智坊7 小时前
多次尝试用 CodeBuddy 做小程序,最终我放弃了
人工智能·ai编程
后端小肥肠8 小时前
别再眼馋 10w + 治愈漫画!Coze 工作流 3 分钟出成品,小白可学
人工智能·aigc·coze
唐某人丶11 小时前
教你如何用 JS 实现 Agent 系统(2)—— 开发 ReAct 版本的“深度搜索”
前端·人工智能·aigc
FIT2CLOUD飞致云11 小时前
九月月报丨MaxKB在不同规模医疗机构的应用进展汇报
人工智能·开源
阿里云大数据AI技术11 小时前
【新模型速递】PAI-Model Gallery云上一键部署Qwen3-Next系列模型
人工智能
袁庭新12 小时前
全球首位AI机器人部长,背负反腐重任
人工智能·aigc
机器之心12 小时前
谁说Scaling Law到头了?新研究:每一步的微小提升会带来指数级增长
人工智能·openai
该用户已不存在12 小时前
Mojo vs Python vs Rust: 2025年搞AI,该学哪个?
后端·python·rust