yolov5 使用engine模型出错

yolov5跑模型的时候 pt模型可以运行 但我使用engine的时候就会报错 _pickle.UnpicklingError: unpickling stack underflow

img

这个错误可能是由于pickle序列化和反序列化过程中的某些问题导致的。您可以尝试以下解决方法：

  1. 确保您正在使用最新版本的PyTorch和YoloV5。

  2. 尝试重新训练模型并保存为新的.pt文件，然后再次尝试使用engine。

  3. 如果您正在使用Python 3.x，请尝试使用Python 2.x运行您的代码。

  4. 将您的模型文件从本地移动到云端，然后再次尝试运行模型。

  5. 尝试使用dill代替pickle进行序列化和反序列化。

  6. 如果您使用的是自定义数据集，请确保您的数据集已正确标记，并且标记文件与图像文件位于同一目录中。

如果您仍然无法解决问题，请考虑向PyTorch或YoloV5社区寻求帮助。

不知道你这个问题是否已经解决, 如果还没有解决的话:

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^

engine怎么会用torch加载，这需要用到tensorRT才对，你的weight参数给的是什么？.engine后缀的话，代码应该是用tensorRT去加载的