rk3588 onnx转rknn报错
（为了高效，请按如下格式提问）硬件主板型号：ROC-3588S-PC
固件名称/系统版本：Ubuntu22.04
自编的或固件下载地址：https://www.t-firefly.com/doc/download/161.html
Log日志: (可以导出，用TXT附件上传)
rknn-toolkit2 version: 2.3.2
onnx==1.18.0
问题描述及复现步骤:
1.使用pytorch的官方的预训练模型resnet50（https://download.pytorch.org/models/resnet50-19c8e357.pth），使用pt2onnx的脚本转化为onnx，测试，识别结果均正常
2.使用瑞芯微官方的onnx转化rknn脚本resnet.py（https://github.com/airockchip/rknn_model_zoo/blob/main/examples/resnet/python/resnet.py）
3.发现报了个错误，虽然rknn的模型导出来了，但是识别结果都是错的
4.然后我用瑞芯微官方提供的onnx(https://ftrg.zbox.filez.com/v2/delivery/data/95f00b0fc900458ba134f8b180b3f7a1/examples/resnet/resnet50-v2-7.onnx)，瑞芯微的脚本转化为rknn,就没有报错，且识别正常。
附件
1.用官方onnx转化没有报错的的log以及 用预训练模型转化为onnx后转化为rknn的报错log
2.pth文件转化为onnx的脚本文件 pt2onnx.py
3.瑞芯微官方转化脚本resnet.py
本帖最后由 风火轮用户oD2dp 于 2025-12-12 09:44 编辑
压缩包的pt2onnx脚本是正确的吗，没有用到下载的权重
风火轮用户oD2dp 发表于 2025-12-12 09:43
压缩包的pt2onnx脚本是正确的吗，没有用到下载的权重
建议重新确认一下，我试了加载下载的权重转换没有报错 感谢回复，我查了下。真的没有把自己训练的权重加载进去。加载进去后，虽然还是会报E RKNN: Unkown op target: 0。但是转化出来的rknn是可以正常使用的。
我自己训练的模型只是改了最后一层的分类头，转化就会报这个错
但是用resnet的预训练模型转成的onnx，或者瑞芯微的提供的onnx就不存在这个报错，不知道是不是什么显示bug px-1 发表于 2025-12-12 16:02
感谢回复，我查了下。真的没有把自己训练的权重加载进去。加载进去后，虽然还是会报E RKNN: ...
rknn推理测试正常其他的就不用管。
页:
[1]