Yolov7-tiny轉檔問題

Timothy

您好，我們將官方的模型yolov7-tiny.pt 透過官方的 export.py 轉成.onnx檔，在用kneron toolchain轉成NEF檔時報以下錯誤(如圖)，想詢問解決辦法，謝謝。

以下連結為附檔

1.yolov7-tiny.py

2.yolov7-tiny.onnx(透過python export.py --weights yolov7-tiny.pt --grid --simplify轉成onnx檔)

3.export.py(轉檔用的script)

https://drive.google.com/drive/folders/1ItLVxJQ1Z13z38iqPppFl6It4tNW3c5n?usp=sharing

Ethon Lin

Hello,

下面幾個檢查方向可供你參考

• pytorch的模型轉成onnx後還需要再跑一次converter，來將模型轉換成符合耐能NPU的格式，詳細可參考耐能文件的3.1.2. Pytorch to ONNX

https://doc.kneron.com/docs/#toolchain/manual_3_onnx/#31-model-conversion-and-optimization

• 模型方面，為了更貼近耐能NPU架構，建議先執行以下修改後再進行訓練

modify train/test input dynamic range to [-0.5,0.5] (耐能NPU為INT8架構)

Use yolov7 variant and reduce the channel to 0.5x

Modified activation function to leaky relu.