ONNX模型转MNN模型中permute使用说明

4 篇文章 0 订阅
3 篇文章 0 订阅

ONNX模型转MNN模型中permute使用说明

问题描述:在利用TensorRT或MNN或OpenVINO等推理引擎加载ONNX模型的过程中,为了少写一些代码,经常会将对图像均值归一化、通道变化等处理写到ONNX模型中。然而在利用MNN推理引擎来加载ONNX模型时,推理结果与原训练模型输出结果不一致,经过排查发现,MNN中的permute函数不支持NCHW或者NHWC格式。MNN中permute使用方式如下:

permute:
Inputs:
• input: float32(NC4HW4)
Outputs:
• output: float32

为解决上述问题,可将通道变换拿到ONNX模型导出的外面。

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值