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

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

img

这里不注释掉没法运行:

img

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

img

指向的代码片段:

img

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

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

您好,我是有问必答小助手,您的问题已经有小伙伴帮您解答,感谢您对有问必答的支持与关注!
PS:问答VIP年卡 【限时加赠:IT技术图书免费领】,了解详情>>> https://vip.csdn.net/askvip?utm_source=1146287632