jsxapril

onnx导出rknn模型输出结果严重错误问题

3

想要导出一个pytorch模型到rknn，因为这个模型融合了两组模型参数（DISTS: Vgg16+自训练权重）所以采用的先导出到onnx模型、再转化成rknn模型的方式；经过验证，测试数据通过onnx模型推理与pytorch模型精度一致（0.309），但再次转化到rknn模型则结果值完全错误（43744），并且经过反复确认输出数据的预处理模式，偏差量级无差别，可能原因是什么，或是如何定位该问题出错的原因？doc文档已看未找到解决方案。 测试环境： onnx 1.6.0 onnxoptimizer 0.3.13 onnxruntime 1.9.0 protobuf 3.11.2 Numpy 1.19.5 torch 1.9.0 torchvision 0.10.0 onnx模型导出代码： # 创建模型实例 model = DISTS() model.eval() # 定义输出文件名 output_onnx = '/data/jishuangxi/workspace/iqa_cpp_version/ref_files/DISTS_ONNX/myself_model2.onnx' output_onnx2...