onnx 模型轉換 rknn 維度計算錯誤問題

大家好

我有訓練一個 pytorch 模型並且想經由RK1808運算棒做計算
目前做法是先將 pytorch 模型轉成 onnx 模型 (opset11) ，再將 onnx 模型轉至 rknn
onnx 模型經由測試資料驗證，轉換與模型輸出結果皆正常而將onnx 轉成 rknn 時，輸出維度錯誤，不符合原本輸出

根據轉換後的結果(如圖所示)
permute輸出(500x150x1x1) 結果符合原始模型，並與 variable (220x500) 做矩陣相乘(MATRIXMUL)
正常結果應為 220 x 150 x 1 x 1
但是轉換後卻輸出 500 x 500 x 1 x 1


請問這部分該如何處理 ?

rknn toolkit 跟 npu driver 皆為 1.7.0

謝謝

先更新1.7.1试下，如果还不行，请提交完整的代码和模型。

bobby_jiang 发表于 2022-4-26 09:49
先更新1.7.1试下，如果还不行，请提交完整的代码和模型。

您好
我有嘗試過使用 1.7.1 結果也是相同
但是如果我把原本的 4 dimension (500 x 150 x 1 x 1)  改成  3 dimension (500 x 150 x 1) ，
這樣結果就會正常，
請問這是甚麼原因呢?
謝謝