我在做一个病虫害识别的的小玩意,用的是YOLO-fastestv2
我根据某位老哥的方法,导出了ONNX文件https://blog.csdn.net/weixin_47296244/article/details/120943801
但是同时我有很多问题
1,我在用它的方法测试使用,命令行可以输出预测时间,但是没有反馈图片
2,我在用这个ONNX文件导入STM32CUBEMX转为底层库函数时候出现INTERNAL ERROR: Shape with 0 value [0, 24, 44, 44],可以看出这个是张量有问题,我试着导出原版检测ONNX给他的但是结果还是一样,我应该怎么解决这个问题
3,我并不知道其格式位,这个CUBEMX只支持INT8和FP32,如果出现格式不匹配我又应该如何处理
![](https://profile-avatar.csdnimg.cn/b2e6b59fe53840e5970474f9d76ad988_bluebeach_wjm.jpg!4)
关于#YOLO#模型导入STM32CUBE的问题,如何解决?
- 写回答
- 好问题 0 提建议
- 追加酬金
- 关注问题
- 邀请回答
-
2条回答 默认 最新
- 爱晚乏客游 2023-07-03 13:31关注
先说格式问题。格式这个可以选择导出对应的格式的onnx,或者应该你用的库有转onnx格式的功能,类似于onnxruntime就可以在推理的时候选择推理的格式。正常来说一般都是FP32的格式,除非你这个格式速度跟不上你的需求才回去考虑int8的问题,不然我建议你先将FP32的运行跑通再去考虑int8的问题。
第二,你的shape报错的意思是第一维度不能为0.也就是你的batch size至少要为1才对,一般图片是三维的,[c,H,W] ,而网络都是4维度的输入,就是多了一个batch size的维度,也就是[N,C,H,W]或者[N,H,W,C],后者是channel 在后的模式,一般是tensorflow用的多,pytorch一般是channel在前。所以你需要检查一下你这里的升维是否正确本回答被题主选为最佳回答 , 对您是否有帮助呢?解决 无用评论 打赏 举报
悬赏问题
- ¥20 需要帮我远程操控一下,运行一下我的那个代码,我觉得我无能为力了
- ¥20 有偿:在ubuntu上安装arduino以及其常用库文件。
- ¥15 请问用arcgis处理一些数据和图形,通常里面有一个根据点划泰森多边形的命令,直接划的弊端是只能执行一个完整的边界,但是我们有时候会用到需要在有很多边界内利用点来执行划泰森多边形的命令
- ¥30 在wave2foam中执行setWaveField时遇到了如下的浮点异常问题,请问该如何解决呢?
- ¥750 关于一道数论方面的问题,求解答!(关键词-数学方法)
- ¥200 csgo2的viewmatrix值是否还有别的获取方式
- ¥15 Stable Diffusion,用Ebsynth utility在视频选帧图重绘,第一步报错,蒙版和帧图没法生成,怎么处理啊
- ¥15 请把下列每一行代码完整地读懂并注释出来
- ¥15 寻找公式识别开发,自动识别整页文档、图像公式的软件
- ¥15 为什么eclipse不能再下载了?