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)
一、你使用的TensorRT版本与YOLOv5版本不一致
用我这组搭配一定是可以的,主要是yolov5的训练模型的工程不匹配,因为是在yolov5的工程里将pt模型转化为wts模型。然后再在tensorrtx里将wts模型转化为engine模型的。第一步骤在yolov5工程里有问题,所以造成在tensorrtx里将模型转化engine有问题。
将我组合通过网盘发布出来:链接: https://pan.baidu