STM32CubeIDE部署神经网络.onnx错误INTERNAL ERROR: Unknown shape destinatio

点击对AI算法分析后出错,如下

STM32CubeIDE部署神经网络.onnx错误INTERNAL ERROR: Unknown shape destinatio_第1张图片

这是PyTorch模型转换到ONNX模型中非常常见的reshape的问题,一个简单的PyTorch中的view操作,期待出来的就是一个简单的reshape操作,但是如果你的形状参数用的是形如x.size(0)这样的话就会出现上图的问题,必须做一个强制类型转换int(x.size(0),或者用onnx-simplifier来处理一下模型,如下:

STM32CubeIDE部署神经网络.onnx错误INTERNAL ERROR: Unknown shape destinatio_第2张图片

推荐文章:利用pytorch搭建指纹识别模型.onnx,并部署到开发板

【嵌入式AI开发】篇五|实战篇一:STM32cubeIDE上部署神经网络之pytorch搭建指纹识别模型.onnx上篇介绍了使用官方提供的STM32cubeIDE部署AI模型的流程,从中可以了嵌入式AI部署的整个流程。本文开始在STM32cubeIDE上进行实战开发训练,本篇主要讲解数据准备和指纹识别模型的搭建与部署。https://mp.weixin.qq.com/s/zPKGctt75Q96xDOPasLEtw

你可能感兴趣的:(STM32开发AI,神经网络,STM32)