請問用docker kneron toolchain轉換模型是否只接受圖片

夏英杰夏英杰
in Innoserve area

你好，我們的onnx模型的輸入只有3個參數，如圖所示。請問耐能的docker toolchain是否只能接受4個參數的輸入?

因為我看 workspace/scripts/utils/img_preprocess.py 中的 img2txt_bin 是要給 out_h 跟 out_w 的，而我在執行 fpAnalyserCompilerIpevaluator_520.py 的時候會有 "list out of range" 的錯誤。

image.png


Comments

  • Ethon LinEthon Lin

    是的，目前toolchain對於模型的輸入格式為 1x height x weight x ch，若是ch數為1的情況下也還是請保持四個維度的狀態

  • 夏英杰夏英杰

    好的，謝謝

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

© Kneron, Inc.