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

相关推荐
MO2T8 分钟前
使用 Flask 构建基于 Dify 的企业资金投向与客户分类评估系统
后端·python·语言模型·flask
慢热型网友.11 分钟前
用 Docker 构建你的第一个 Python Flask 程序
python·docker·flask
Naiva11 分钟前
【小技巧】Python + PyCharm 小智AI配置MCP接入点使用说明(内测)( PyInstaller打包成 .exe 可执行文件)
开发语言·python·pycharm
云动雨颤15 分钟前
Python 自动化办公神器|一键转换所有文档为 PDF
运维·python
moonless022221 分钟前
🌈Transformer说人话版(二)位置编码 【持续更新ing】
人工智能·llm
小爷毛毛_卓寿杰21 分钟前
基于大模型与知识图谱的对话引导意图澄清系统技术解析
人工智能·llm
梅孔立28 分钟前
yum update 报错 Cannot find a valid baseurl for repo: centos-sclo-rh/x86_64 等解决办法
linux·python·centos
聚客AI32 分钟前
解构高效提示工程:分层模型、文本扩展引擎与可视化调试全链路指南
人工智能·llm·掘金·日新计划
前端付豪44 分钟前
13、你还在 print 调试🧾?教你写出自己的日志系统
后端·python
摆烂工程师1 小时前
Claude Code 落地实践的工作简易流程
人工智能·claude·敏捷开发