一、Parameter check failed at: ../builder/Network.cpp::addScale::482, condition: shift.count > 0

出现这个问题是因为模型的yolo网络版本与加速过后的yolo网络版本不一致。
比如我的模型是yolov5-5.0,然后我git了yolov5-6.0的tensorrt加速项目,那么就会出现这种报错

解决方案:
删除已经克隆的yolov5-6.0加速项目,重新克隆yolov5-5.0加速项目

二、想要获取的参数数量不对xx expected but xx get

这是因为在执行cmke ..命令之前,需要修改yololayer.h文件中的CLASS_NUM参数,这个参数默认是80(coco数据集中的类别数量),如果是在自己的数据集上训练的模型,需要将CLASS_NUM参数修改为自己的类别数量,我的数据集只有一个类别,所以将该参数改成1。

【重要!!!!】如果你没有修改CLASS_NUM参数就执行了cmake ..以及make命令,然后出现了该错误,那么你需要先修改CLASS_NUM参数,然后把build文件删掉,重新按mkdir build``cd build``cmake ..``make顺序执行命令,因为cmake生成的文件是依赖yololayer.h文件的,修改了该文件中的参数就要重新cmake