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层即可。