yolov8转snpe时Concat报错问题解决方法
yolov8转snpe时Concat报错问题解决方法
整个问题的引起是我需要将yolov8部署到高通的8155芯片上，因此需要将pytorch模型转换到snpe格式。大体流程是：pytorch->onnx->dlc->量化。
整个过程遇到了两个奇怪问题：
1、转onnx时输入必须是‘inputx’，而yolov8默认是‘images’, 这个问题需要在exporter.py中修改；
2、使用snpe1.59.0.3230版本，当Concat的两个输入是Mul和Sigmoid时，onnx转dlc时会报错。主要看输出信息，以警告的形式给出，但是量化时会报错。
解决方法时在Mul算子后面增加一个reshape层即可。
原文地址：https://blog.csdn.net/Sunshine_in_Moon/article/details/137689063
免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！