pth转onnx报错：AttributeError: 'collections.OrderedDict' object has no attribute 'shape'

Question

使用的代码是：https://github.com/hpc203/yolov5-dnn-cpp-python
里的convert_onnx.py，我修改了一些部分：

这里不注释掉没法运行：

可以稍微运行一会儿，但是又很快报错：

指向的代码片段：

不知道这个“shape”到底是哪里出了问题，是因为我只有一个类吗？

Answer 1

从报错看，是因为utl_param.values()和own_state.values()这些变量是odict类型，无shape属性，检查这两个变量，将其转换成np的ndarray或pd的Series才可用shape