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

[复制链接] · 发表于 2022-4-6 15:04:55

本帖最后由 arthurwu 于 2022-4-15 11:40 编辑

大家好

我有訓練一個 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

謝謝

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

本帖子中包含更多资源