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