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+深度学习领域博主

猜你喜欢

转载自blog.csdn.net/qq_58355216/article/details/124680092