您好!
我将创建一个定制电路板、并在 SDK 9.0中使用 EdgeAI ModelMaker Repo R9.0对对象检测模型进行原型设计。 我能够顺利地培训每个 YOLO 模型并进行部署。 为了解决一些不相关的驱动程序问题、几周前我必须升级到 SDK 9.1.7版。 我刚刚切换到了 EdgeAI ModelMaker 的正确分支(r9.1),现在在运行推理时出现此问题:
2022年04月30日19:59:53.738820485 [W:onnxruntime:、execution_frame.cc:835 VerifyOutputSizes]来自{-1}模型的预期形状与输出标签的{1、1、1、11200}实际形状不匹配
2022年04月30日19:59:53.749074035 [W:onnxruntime:、execution_frame.cc:835 VerifyOutputSizes]来自{-1、5}模型的预期形状与输出设计的实际形状{1、1、1200、5}不匹配
如果我禁用 TIDL 执行、模型在 CPU 上运行缓慢、没有错误。 这似乎指向编译/人工制品作为问题。 如果我从 allowedNode.txt (283、我认为)中删除输出节点、它也可以正常运行-速度太慢了、不符合我的要求。
输出形状打印为:
NodeArg(name='dets',type='tensor (float)',shape =['Concat1482_dim_0',5])。 参考层的尺寸似乎未正确填充。
我将 tarball 附加为 modelMaker 的输出。 如果您对正在发生的事情有任何想法、我将非常感谢您的帮助。
谢谢!