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主函数中的参数
修改完之后就可以了,稍等一会就能生成了
感谢大佬的帮助,在此放上他的链接: