Model convert的疑問

想請問使用toolchain中的ONNX to ONNX(http://doc.kneron.com/docs/#toolchain/converters/#6-onnx-to-onnx-onnx-optimization)轉換出的onnx model會有

的錯誤訊息。

model的架構:

此model是使用https://github.com/onnx/tensorflow-onnx把tensorflow2的model轉換成onnx model生成的,傳換時opset設定為11。

請問這個問題該如何解決,謝謝。

Comments

  • @Kuan

    Hi Kuan,

    看了您的錯誤訊息,可能是在您的 script 裡面,您的input dimensions 設置錯誤。

    您可以確認一下您的 input_image input size 是什麼,確認完成之後檢查您的 script 內的 input dimensions 是否符合您模型 input size。

  • 改為正確的input size後發生了batchsize的問題,將model的input size的batchsize改為1還是發生相同的錯誤,不知道是不是有哪些沒有注意到的細節。

    另外請問如果使用ktc.onnx_optimizer.tflite2onnx_flow指令來轉換tflite的model有辦法像h5一樣指定input shape嗎


    謝謝

  • @Kuan

    Hi Kuan,

    因為您提供的資訊不太足夠,我不好判斷,所以是否可以提供您跑的 scripts 還有 model。

    回答您的問題,如果您使用 ktc.onnx_optimizer.tflite2onnx_flow 指令來轉換 tflite 的 model 是沒有辦法像 h5 一樣指定 input shape 的。

  • 謝謝,上面的問題我自行解決了。

    目前在轉換另一個model的時候在km.analysis的時候有發生錯誤

    轉換的model架構如下圖

    在onnx optimization時有加上--add-bn -t --opt-matmul --no-bn-fusion等參數。

    想請問是model的架構有問題還是在轉換的時候哪個步驟做錯了,謝謝。

  • @Kuan

    Hi Kuan,

    看了您的錯誤訊息與 model 架構,造成 km.analysis 錯誤原因,初步判斷可能是因為 model 的架構有不符合硬體的架構 (ex. Reshape、Transpose等),或是在轉換 model 沒有成功的 onnx optimize 導致錯誤。

    不過,您提供的資訊不好判斷具體情況,可以的話,還請您提供一下您跑的 scripts 還有 model,讓我們能排除錯誤原因,非常感謝!

The discussion has been closed due to inactivity. To continue with the topic, please feel free to post a new discussion.