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

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