yolov5 onnx轉nef
想請問一下，我是用yolov5的export.py 轉成onnx，並將opset設置成12，然後照著官網的流程去做ktc.onnx_optimizer.onnx2onnx_flow()，但是最後結果還是opset的問題。
想知道這個部分要如何去解決，感謝。
The discussion has been closed due to inactivity. To continue with the topic, please feel free to post a new discussion.
Comments
您好，
可以請您提供您使用的onnx模型嗎? 謝謝!
這是我經過ktc.onnx_optimizer.onnx2onnx_flow()之後的onnx
這是經過export.py的onnx檔
Hello,
模型後段看起來有些layer維度超過NPU的限制了，NPU只能接受4維 batch x channel x height x width，且batch size必須要是1
建議照紅線處裁剪掉模型後段的layers，並將裁剪掉的部分改在後處理內實現(請務必記得加入此部分運算，否則模型結果會錯誤)
若想使用的模型為yolo的話，有關yolov5或yolov7的部分，耐能開發者網站上有提供一份編修指南也提供給您參考https://www.kneron.com/tw/support/developers/?folder=Documentation%20center/&download=1904
您好，我按照編修指南重新訓練一個 yolov5 並轉成 onnx ，想請您們看一下這樣子是正確的嗎 ?
?
這是我經過ktc.onnx_optimizer.onnx2onnx_flow()之後的onnx
@黃浩天
Hi 浩天,
以 onnx 模型結構來說 (from onnx2onnx_leakyRelu.zip) 您所提供的 onnx 是正確並且可在 KL720 上所使用。