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

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模型导出的外面。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章