TensorRT生成engine文件异常（[TRT] Network::addScale::434, condition: shift.count ＞ 0 ?)

Loading weights: ./best.wts
[05/10/2022-08:41:14] [E] [TRT] Parameter check failed at: Network.cpp::nvinfer1::Network::addScale::434, condition: shift.count > 0 ? (shift.values != nullptr) : (shift.values == nullptr)

这个错误是wts转engine文件时的报错，在网上找了好久也没有找到靠谱的解决方法

后来经过大佬的帮助成功解决，接下来就介绍可能引发此异常的原因

一、

你使用的TensorRT版本与YOLOv5版本不一致

二、

yololayer.h文件中类别和图片大小没修改（图片大小是你训练时img-size的大小）

三、

修改yolov5T.cpp主函数中的参数

修改完之后就可以了，稍等一会就能生成了

​​​​​​​

 

感谢大佬的帮助，在此放上他的链接

Mr曲末寒的博客_CSDN博客-emgucv-图像处理+深度学习,AI-训练+部署,TensorRT+深度学习领域博主