为什么我用yolov5-master中的export.py脚本将我训练好的pt模型转为onnx后,我去测试转化好的onnx模型时,会报这样的错误

Viewed 80

期待结果和实际结果

软硬件版本信息


错误日志

尝试解决过程


这是我在pycharm终端尝试跑一下我的onnx模型
补充材料

1 Answers

你查一下输入数据是否正确

我已经解决了,我练的模型的图片尺寸是640X640的,转onnx是320X320的,尺寸上不一致出现了问题。感谢你的建议。