yolov5中pt模型转onnx模型报错_运行yolo export.py attributeerror: 'list' object has

作者：从前慢现在也慢 | 2024-07-19 07:30:17

踩

运行yolo export.py attributeerror: 'list' object has no attribute 'shape

yolov5-7.0中.pt文件转onnx文件时候修改yolo.py文件中修改class Detect中的forward函数为
def forward(self, x):
z = [] # inference output
for i in range(self.nl):
x[i] = self.m[i](x[i]) # conv
return x
后用export.py脚本转onnx模型报错：AttributeError: 'list' object has no attribute 'shape'

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/从前慢现在也慢/article/detail/850132